Update app.py

app.py

@@ -139,7 +139,7 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
         height=height,
         num_images_per_prompt=num_images_per_prompt,
         generator=generator
-    ).images[0]
+    ).images
     
     return image, seed
 
@@ -219,7 +219,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="blue", secondar
         
         with gr.Column(scale=1):
             with gr.Group(elem_classes="output-group"):
-                output_image = gr.Image(label="Generated Image")
+                output_image = gr.Gallery(label="Result", elem_id="gallery", show_label=False)
                 final_prompt = gr.Textbox(label="Final Prompt Used")
                 used_seed = gr.Number(label="Seed Used")