Update app.py

app.py

@@ -1,6 +1,4 @@
 import gradio as gr 
-
-
 import torch
 from utils import get_image_from_url, colorize
 from PIL import Image
@@ -17,10 +15,10 @@ 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)
 
-    return depth_pil
+    return colored_depth
 
 interface = gr.Interface(fn=process_image, 
                      inputs=[gr.Image(type="pil")],