lemonaddie commited on
Commit
4ea140d
1 Parent(s): a0183ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- 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,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