MaxMilan1 commited on
Commit
9dfb708
·
2 Parent(s): c55a0c7 22eb08b

Merge branch 'main' of https://huggingface.co/spaces/2MaxM/ShoeGen

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -24,7 +24,9 @@ with gr.Blocks(_TITLE) as ShoeGen:
24
  prompt = gr.Textbox(label="Enter a prompt")
25
  button_gen = gr.Button("Generate Image")
26
  with gr.Column():
27
- image = gr.Image(label="Generated Image", show_download_button=True)
 
 
28
 
29
  button_gen.click(generate_image, inputs=[prompt], outputs=image)
30
 
 
24
  prompt = gr.Textbox(label="Enter a prompt")
25
  button_gen = gr.Button("Generate Image")
26
  with gr.Column():
27
+ # show images
28
+ gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery", show_download_button=True, columns=[2])
29
+
30
 
31
  button_gen.click(generate_image, inputs=[prompt], outputs=image)
32