Spaces:
Running
on
Zero
Running
on
Zero
lemonaddie
commited on
Commit
•
4ea140d
1
Parent(s):
a0183ba
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,8 @@ from PIL import Image
|
|
11 |
|
12 |
from gradio_imageslider import ImageSlider
|
13 |
|
|
|
|
|
14 |
def process(
|
15 |
pipe,
|
16 |
path_input,
|
@@ -488,7 +490,7 @@ def run_demo_server(pipe):
|
|
488 |
|
489 |
|
490 |
|
491 |
-
|
492 |
def main():
|
493 |
|
494 |
REPO_URL = "https://github.com/lemonaddie/geowizard.git"
|
@@ -501,7 +503,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 +517,6 @@ def main():
|
|
515 |
|
516 |
pipe = pipe.to(device)
|
517 |
|
518 |
-
@spaces.GPU
|
519 |
run_demo_server(pipe)
|
520 |
|
521 |
|
|
|
11 |
|
12 |
from gradio_imageslider import ImageSlider
|
13 |
|
14 |
+
import spaces
|
15 |
+
|
16 |
def process(
|
17 |
pipe,
|
18 |
path_input,
|
|
|
490 |
|
491 |
|
492 |
|
493 |
+
@spaces.GPU
|
494 |
def main():
|
495 |
|
496 |
REPO_URL = "https://github.com/lemonaddie/geowizard.git"
|
|
|
503 |
repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
|
504 |
sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
|
505 |
|
506 |
+
|
507 |
from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
|
508 |
|
509 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
517 |
|
518 |
pipe = pipe.to(device)
|
519 |
|
|
|
520 |
run_demo_server(pipe)
|
521 |
|
522 |
|