lemonaddie commited on
Commit
a0183ba
1 Parent(s): cfc0ae6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- +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 +515,7 @@ def main():
515
 
516
  pipe = pipe.to(device)
517
 
518
- +@spaces.GPU
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