Update app.py

app.py

@@ -123,7 +123,7 @@ def generate(
         negative_prompt = None  # type: ignore
     prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
 
-    images = pipe(
+    image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
         width=width,
@@ -131,14 +131,12 @@ def generate(
         guidance_scale=0,
         num_inference_steps=inference_steps,
         generator=generator,
-        num_images_per_prompt=NUM_IMAGES_PER_PROMPT,
         use_resolution_binning=use_resolution_binning,
-        output_type="pil",
-    ).images
+    ).images[0]
 
     image_paths = [save_image(img) for img in images]
     print(image_paths)
-    return image_paths, seed
+    return images, seed
 
 
 examples = [
@@ -172,7 +170,7 @@ with gr.Blocks(css=css) as demo:
                     container=False,
                 )
                 run_button = gr.Button("Run", scale=0)
-            result = gr.Gallery(label="Result", columns=1,  show_label=False)
+            result = gr.Image(label="Result")
                     
     with gr.Accordion("Advanced options", open=False):
         with gr.Group():