Update app.py

app.py

@@ -46,9 +46,15 @@ def process_image(image_path):
       Image.fromarray(depth_image)
     )
 
-   
 
-    return depth_image_padded
+        # 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
+
 
 
 examples_images = [[f] for f in sorted(glob.glob("examples/*.jpg"))]
@@ -69,7 +75,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.Image(label="Predicted Depth", type="pil")
+            predicted_depth = gr.File(label="Predicted Depth (Download PNG)")
     with gr.Row():
         autostereogram = gr.Image(label="Autostereogram", type="pil")
     with gr.Row():