Update app.py

app.py

@@ -52,19 +52,17 @@ def process_image(image: Image.Image):
     image = image.convert("RGB")
 
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    # model.to(device)
-    # depth = model.infer_pil(image)
-
-    # processed_array = save_raw_16bit(colorize(depth)[:, :, 0])
-    # return Image.fromarray(processed_array)
-
     model.to(device)
-    
-    # # inference
-    processed_array = pipe(image)["depth"]
+    depth = model.infer_pil(image)
 
+    processed_array = save_raw_16bit(colorize(depth)[:, :, 0])
     return Image.fromarray(processed_array)
 
+    # model.to(device)
+    
+    # processed_array = pipe(image)["depth"]
+    # return Image.fromarray(processed_array)
+
 # ----------- Depth functions
 
 # ----------- Mesh functions