Update UI

app.py

@@ -61,7 +61,7 @@ def infer(
     print(prompt)
     print(negative_prompt)
 
-    image = pipe(
+    images = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
         guidance_scale=guidance_scale,
@@ -70,9 +70,9 @@ def infer(
         height=image_size,
         generator=generator,
         num_images_per_prompt=num_images,
-    ).images[0]
+    ).images
 
-    return image, seed
+    return images
 
 css = """
 #col-container {
@@ -120,8 +120,8 @@ with gr.Blocks(css=css) as demo:
             )
 
         run_button = gr.Button("Run", scale=0, variant="primary")
-        # result = gr.Image(label="Result", show_label=False)
-        gallery_result = gr.Gallery(label="Generated Images", show_label=False)
+        #result = gr.Image(label="Result", show_label=False)
+        result = gr.Gallery(label="Generated Images", show_label=False)
 
 
         with gr.Accordion(open=False, label="Advanced Options"):
@@ -183,7 +183,7 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps,
             num_images
         ],
-        outputs=[gallery_result],
+        outputs=[result],
     )
 
 if __name__ == "__main__":