MaxMilan1 commited on
Commit
b684483
·
1 Parent(s): 9dfb708
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ with gr.Blocks(_TITLE) as ShoeGen:
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)
 
25
  button_gen = gr.Button("Generate Image")
26
  with gr.Column():
27
  # show images
28
+ image = gr.Image(label="Generated Image", show_download_button=True)
29
 
30
 
31
  button_gen.click(generate_image, inputs=[prompt], outputs=image)