Update app.py

app.py

@@ -31,7 +31,6 @@ def process_image(image):
     output = prediction.cpu().numpy()
     formatted = (output * 255 / np.max(output)).astype('uint8')
     img = Image.fromarray(formatted)
-    
     return img
 
 # Interface
@@ -41,7 +40,7 @@ description = "Intel's DPT: Dense Prediction Transformer for depth estimation fr
 iface = gr.Interface(
     fn=process_image, 
     inputs=gr.inputs.Image(type="pil", label="Input Image"),
-    outputs=gr.outputs.Image(type="pil", label="Predicted Depth"),
+    outputs=predicted_depth,
     title=title,
     description=description,
     allow_flagging="never"