lemonaddie commited on
Commit
aa7a82a
1 Parent(s): 4018a14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -406,15 +406,15 @@ def main():
406
 
407
  REPO_URL = "https://github.com/lemonaddie/geowizard.git"
408
  CHECKPOINT = "lemonaddie/Geowizard"
409
- REPO_DIR = "pipeline"
410
-
411
  if os.path.isdir(REPO_DIR):
412
  shutil.rmtree(REPO_DIR)
413
 
414
  repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
415
  sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
416
 
417
- from .models.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
418
  pipeline = DepthNormalEstimationPipeline.from_pretrained("lemonaddie/Geowizard")
419
 
420
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
406
 
407
  REPO_URL = "https://github.com/lemonaddie/geowizard.git"
408
  CHECKPOINT = "lemonaddie/Geowizard"
409
+ REPO_DIR = "geowizard"
410
+
411
  if os.path.isdir(REPO_DIR):
412
  shutil.rmtree(REPO_DIR)
413
 
414
  repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
415
  sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
416
 
417
+ from pipeline.models.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
418
  pipeline = DepthNormalEstimationPipeline.from_pretrained("lemonaddie/Geowizard")
419
 
420
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")