Update app.py

app.py

@@ -21,9 +21,8 @@ img {
 }
 """
 
-# DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
-DEVICE = 'cuda'
-model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to("cpu").eval()
+DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
+model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to(DEVICE).eval()
 
 CHECKPOINT = "prs-eth/marigold-v1-0"
 pipe = MarigoldPipeline.from_pretrained(CHECKPOINT)
@@ -45,8 +44,7 @@ def process_image(image: Image.Image):
     global model
     image = image.convert("RGB")
 
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    model.to(device)
+    # model.to(DEVICE)
     depth = model.infer_pil(image)
 
     processed_array = save_raw_16bit(colorize(depth)[:, :, 0])