still same crash, use download button for dummy

app.py

@@ -13,16 +13,17 @@ def dummy_damage_classification(img, threshold):
     # Convert to numpy array
     image_np = np.array(img)
 
-    # Dummy: draw a red square on the image
+    # Draw a red square on the image
     image_pil = Image.fromarray(image_np).convert("RGB")
     draw = ImageDraw.Draw(image_pil)
     draw.rectangle([10, 10, 50, 50], outline="red", width=3)
     image_pil.save(IMAGE_PATH)
 
-    # Dummy CSV content
+    # Create dummy CSV file
     df = pd.DataFrame({"x": [20], "y": [30], "damage_type": ["Dummy"]})
     df.to_csv(CSV_PATH, index=False)
 
+    # Return PIL image for display, and file paths (strings) for download buttons
     return image_pil, IMAGE_PATH, CSV_PATH
 
 with gr.Blocks() as app:
@@ -32,14 +33,14 @@ with gr.Blocks() as app:
     threshold_input = gr.Number(value=20, label="Threshold")
 
     output_image = gr.Image(label="Classified Image")
-    download_image = gr.File(label="Download Image")
-    download_csv = gr.File(label="Download CSV")
+    download_image_btn = gr.DownloadButton(label="Download Image")
+    download_csv_btn = gr.DownloadButton(label="Download CSV")
 
     btn = gr.Button("Run Classification")
     btn.click(
         dummy_damage_classification,
         inputs=[image_input, threshold_input],
-        outputs=[output_image, download_image, download_csv]
+        outputs=[output_image, download_image_btn, download_csv_btn]
     )
 
 if __name__ == "__main__":