Update app.py

app.py

@@ -108,7 +108,7 @@ def generate(
             num_inference_steps = num_inference_steps,
             generator = generator,
             output_type="pil",
-        ).images[0]
+        ).images
 
         return img, seed
 
@@ -196,12 +196,14 @@ with gr.Blocks(
 
         run_button = gr.Button("Run")
 
-        result = gr.Image(
-            label="Result",
+        result = gr.Gallery(
+            label="Result", 
+            columns=1, 
+            height="512px", 
+            preview=True, 
             show_label=False
         )
 
-    with gr.Group():
         used_seed = gr.Number(label="Used Seed", interactive=False)
 
     gr.on(