Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = "
|
| 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")
|