Update app.py

app.py

@@ -619,7 +619,8 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             # gallery = gr.Gallery(label="Generated Images")
             gallery = gr.Gallery(label="Generated Images", show_share_button=False, show_download_button=False)
-            download_files = gr.Files(label="Download Generated Images", file_types=[".png"])
+            download1 = gr.File(label="Download Image 1 (PNG)")
+            download2 = gr.File(label="Download Image 2 (PNG)")
             
             # num_outputs = gr.Slider(label="Select Image Index", minimum=0, maximum=1, step=1, value=0)
             # download_btn = gr.DownloadButton(label="Download Selected Image", elem_id="download_btn")