lemonaddie commited on
Commit
904d41c
1 Parent(s): 9c130b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -406,9 +406,13 @@ def main():
406
 
407
  REPO_URL = "https://github.com/lemonaddie/geowizard.git"
408
  CHECKPOINT = "lemonaddie/Geowizard"
 
409
 
410
- repo = git.Repo.clone_from(REPO_URL)
411
- sys.path.append(os.path.join(os.getcwd()))
 
 
 
412
 
413
  from models.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
414
  pipeline = DepthNormalEstimationPipeline.from_pretrained("lemonaddie/Geowizard")
 
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 models.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
418
  pipeline = DepthNormalEstimationPipeline.from_pretrained("lemonaddie/Geowizard")