Spaces:
Runtime error
Runtime error
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 |
|