Update app.py

app.py

@@ -500,11 +500,11 @@ def main():
  
  repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
  sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
- 
+
+ +import spaces
  from pipeline.depth_normal_pipeline_clip_cfg import DepthNormalEstimationPipeline
 
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
- a = torch.zeros((1), device=device)
  pipe = DepthNormalEstimationPipeline.from_pretrained(CHECKPOINT)
  
  try:
@@ -514,6 +514,8 @@ def main():
  pass # run without xformers
 
  pipe = pipe.to(device)
+ 
+ [email protected]
  run_demo_server(pipe)