Update app.py

app.py

@@ -14,9 +14,9 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 zoe = model_zoe_n.to(DEVICE)
 
 def process_image(image):
-    depth_numpy = zoe.infer_pil(image)  # as numpy
-    depth_pil = zoe.infer_pil(image, output_type="pil")  # as 16-bit PIL Image
-    colored_depth = colorize(depth_pil)
+    depth = zoe.infer_pil(image)  # as numpy
+
+    colored_depth = colorize(depth)
 
     return colored_depth