Update app.py

app.py

@@ -130,7 +130,7 @@ def generate(
             generator = generator,
             num_images_per_prompt=1,
             output_type="pil",
-        ).images
+        ).images[0]
 
         return img, seed
 
@@ -254,11 +254,8 @@ with gr.Blocks(
 
         run_button = gr.Button("Run")
 
-        result = gr.Gallery(
-            label="Result", 
-            columns=1, 
-            height="512px", 
-            preview=True, 
+        result = gr.Image(
+            label="Result",
             show_label=False
         )
     with gr.Group():