Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
a0183ba
1
Parent(s):
cfc0ae6
Update app.py
Browse files
app.py
CHANGED
@@ -501,7 +501,7 @@ def main():
|
|
501 |
repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
|
502 |
sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
|
503 |
|
504 |
-
|
505 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
506 |
|
507 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
@@ -515,7 +515,7 @@ def main():
|
|
515 |
|
516 |
pipe = pipe.to(device)
|
517 |
|
518 |
-
|
519 |
run_demo_server(pipe)
|
520 |
|
521 |
|
|
|
501 |
repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
|
502 |
sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
|
503 |
|
504 |
+
import spaces
|
505 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
506 |
|
507 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
515 |
|
516 |
pipe = pipe.to(device)
|
517 |
|
518 |
+
@spaces.GPU
|
519 |
run_demo_server(pipe)
|
520 |
|
521 |
|