change depth output

app.py

@@ -50,7 +50,7 @@ def calculate_depth(model_type, img):
     out_im = Image.fromarray(formatted)
     out_im.save("Images/Input-Test/0_d.jpg", "JPEG")
 
-    return f'Images/Input-Test/0_d.jpg'
+    return out_im
 
 def wiggle_effect(slider):
 
@@ -68,7 +68,7 @@ with gr.Blocks() as demo:
     inp = [gr.inputs.Dropdown(midas_models, default="MiDaS_small", label="Depth estimation model type")]
     with gr.Row():
         inp.append(gr.Image(type="pil", label="Input"))
-        out = gr.Image(type="file", label="depth_estimation")
+        out = gr.Image(type="pil", label="depth_estimation")
     btn = gr.Button("Calculate depth")
     btn.click(fn=calculate_depth, inputs=inp, outputs=out)