Update app.py

app.py

@@ -47,13 +47,8 @@ def process_image(image_path):
     )
 
 
-        # Save to temp file as PNG
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
-    depth_image_padded.save(temp_file.name, format="jpg")
-    temp_file.close()
-
     # Return as downloadable file
-    return temp_file.name
+    return depth_image_padded
 
 
 
@@ -75,7 +70,7 @@ to generate the autostereogram (Magic Eye)
             input_image = gr.Image(type="filepath", label="Input Image")
             button = gr.Button("Predict")
         with gr.Column():
-            predicted_depth = gr.File(label="Predicted Depth (Download PNG)")
+            predicted_depth = gr.Image(label="Predicted Depth", type="pil")
     with gr.Row():
         autostereogram = gr.Image(label="Autostereogram", type="pil")
     with gr.Row():