Update app.py

app.py

@@ -1,15 +1,16 @@
 from diffusers import StableDiffusionPipeline, DiffusionPipeline
 import torch
 import gradio as gr
+
 import spaces
 
 css = """
 #img-display-output {
-    max-height: 50vh;
+    max-height: 60vh;
 }
 
 #img-display-output *{
-    max-height: 50vh;
+    max-height: 60vh;
 }
 """
 
@@ -22,8 +23,9 @@ pipe.to("cuda")
 @spaces.GPU(enable_queue=True)
 def generate_sprite(prompt):
     # pipe = pipe.to(DEVICE)
-    image = pipe(prompt).images[0]
-    return image
+    # image = pipe(prompt).images[0]
+    images = pipe(prompt).images
+    return images
 
 title = "# SD_PixelArt_SpriteSheet_Generator"
 description = """Pixel Art Sprite Sheet Generator with Stable Diffusion Checkpoint."""
@@ -34,7 +36,8 @@ with gr.Blocks(css=css) as API:
         
     with gr.Column():
         inputs=gr.TextArea(label="Prompt", placeholder="Prompt")
-        outputs=gr.Image(label="Ouput Image", type='pil', elem_id="img-display-output")
+        # outputs=gr.Image(label="Ouput Image", type='pil', elem_id="img-display-output")
+        outputs=gr.Gallery(label="Ouput Images", columns=4, elem_id="img-display-output")
     generate_btn = gr.Button(value="Generate")
     generate_btn.click(generate_sprite, inputs=inputs, outputs=outputs, api_name="generate_mesh")