Update app.py

app.py

@@ -90,10 +90,13 @@ def save_all_images(images):
     if len(images) == 0:
         return None, None
     zip_filename = create_zip(images)  # Create a zip file from the list of image files
+    gr.Button(link="/file=" + zip_filename)
+
+    # remove?
     zip_base64 = encode_file_to_base64(zip_filename)  # Encode the zip file to base64
     download_link = f'<a href="data:application/zip;base64,{zip_base64}" download="{zip_filename}">Download All</a>'
-    gr.File(download_link)
-    print(download_link)
+    #gr.File(download_link)
+    #print(download_link)
     return zip_filename, download_link
         
 # Function to handle "Save All" button click