Upload app.py with huggingface_hub

app.py

@@ -143,7 +143,7 @@ with gr.Blocks(title="KiwiMix XL v3") as demo:
                             height = gr.Slider(768, 1344, value=1024, step=64, label="Height")
                     run = gr.Button("Generate", variant="primary")
                 with gr.Column(scale=1):
-                    result = gr.Image(label="Result", type="pil", format="png")
+                    result = gr.Image(label="Result", type="pil", format="png", buttons=["download", "fullscreen"])
                     used_seed = gr.Number(label="Seed used", interactive=False)
 
             gr.Examples(
@@ -163,7 +163,7 @@ with gr.Blocks(title="KiwiMix XL v3") as demo:
         with gr.Tab("Global gallery", elem_id="gallery-tab") as gallery_tab:
             gr.Markdown(
                 "Everything generated in this Space, by anyone. Click an image, then use the "
-                "download icon to save it."
+                "download icon to save it, or Download all to grab the whole page."
             )
             refresh = gr.Button("🔄 Refresh gallery")
             gallery = gr.Gallery(
@@ -171,17 +171,9 @@ with gr.Blocks(title="KiwiMix XL v3") as demo:
                 columns=4,
                 height=560,
                 object_fit="cover",
-                show_download_button=True,
                 preview=False,
+                buttons=["download", "download_all", "fullscreen"],
             )
-            download = gr.File(label="Download selected image", interactive=False, visible=False)
-
-            def select_image(evt: gr.SelectData):
-                """Expose the selected gallery image as a downloadable file."""
-                url = evt.value["image"]["url"] if isinstance(evt.value, dict) else evt.value
-                return gr.update(value=url, visible=True)
-
-            gallery.select(select_image, None, download)
             refresh.click(load_gallery, None, gallery)
             gallery_tab.select(load_gallery, None, gallery)