kjghjf

app.py

@@ -12,8 +12,8 @@ pipe.to("cuda")
 @spaces.GPU
 def generate_image(prompt):
     
-    images = pipe(prompt)
-    return images[0]
+    images = pipe(prompt).images
+    return images
 
 _TITLE = "Shoe Generator"
 with gr.Blocks(_TITLE) as ShoeGen:
@@ -22,7 +22,8 @@ with gr.Blocks(_TITLE) as ShoeGen:
             prompt = gr.Textbox(label="Enter a prompt")
             button_gen = gr.Button("Generate Image")
         with gr.Column():
-            output = gr.Image(label="Generated Image")
+        #  show images
+            output = gr.Image("Generated Image", show_download_button=True)
 
     button_gen.click(generate_image, inputs=[prompt], outputs=output)