Spaces:
Running
on
Zero
Running
on
Zero
lemonaddie
commited on
Commit
•
904d41c
1
Parent(s):
9c130b3
Update app.py
Browse files
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 |
-
|
411 |
-
|
|
|
|
|
|
|
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")
|