Update app.py

app.py

@@ -68,6 +68,6 @@ with gr.Blocks() as app:
 
         # Step 6: Download button for the final collage as PNG
         download_button = gr.Button("Download Collage (PNG)")
-        download_button.click(create_collage, inputs=[uploaded_files, grid_layout], outputs=output)
+        download_button.click(create_collage, inputs=[uploaded_files, grid_layout], outputs=gr.File(label="Download Collage"))
 
 app.launch(share=True)  # Public link to share the app