Update app.py

app.py

@@ -63,7 +63,7 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
 
     images = images.reshape((num_samples,) + images.shape[-3:])
     images = pipeline.numpy_to_pil(images)
-    return images[0]
+    return images
 
 examples = [
     ["A watercolor painting of a bird"],
@@ -274,7 +274,7 @@ with block as demo:
                     )
                 btn = gr.Button("Generate image", label="Primary Button", variant="primary")
 
-        gallery = gr.Image(
+        gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
         ).style(height="auto")