Update app.py

app.py

@@ -82,19 +82,19 @@ 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)
+        # 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)
         return depth, resize_image
 
     @(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")
+        # 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
-        moge_model = moge_model.to(DEVICE)
+        # global moge_model
+        # moge_model = moge_model.to(DEVICE)
         metric_depth, mask, intrinsics = moge_model.infer(image)
         metric_depth[~mask] = metric_depth[mask].max()
         return metric_depth, mask, intrinsics