Update app.py

app.py

@@ -47,7 +47,7 @@ css = """
 
 set_seed(666)
 
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+# DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 default_steps = 10
 model = PixelPerfectDepth(sampling_steps=default_steps)
@@ -74,6 +74,7 @@ def main(share=True):
 
     @(spaces.GPU if HUGGINFACE_SPACES_INSTALLED else (lambda x: x))
     def predict_depth(image, denoise_steps):
+        DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         global model
         model = model.to(DEVICE)
         depth, resize_image = model.infer_image(image, sampling_steps=denoise_steps)
@@ -81,6 +82,7 @@ def main(share=True):
 
     @(spaces.GPU if HUGGINFACE_SPACES_INSTALLED else (lambda x: x))
     def predict_moge_depth(image):
+        DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         image = torch.tensor(image / 255, dtype=torch.float32, device=DEVICE).permute(2, 0, 1)
         global moge_model