kgjk

app.py

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