jhf

app.py

@@ -23,7 +23,7 @@ with gr.Blocks(_TITLE) as ShoeGen:
             button_gen = gr.Button("Generate Image")
         with gr.Column():
         #  show images
-            gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
+            gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery", show_download_button=True, columns=[2])
     
 
     button_gen.click(generate_image, inputs=[prompt], outputs=gallery)