Update app.py

app.py

@@ -9,12 +9,12 @@ pipe = StableDiffusionPipeline.from_pretrained(
 ).to(device)
 
 def generate(prompt, steps, guidance, seed):
-    gen = None if seed == 0 else torch.manual_seed(seed)
-    image = pipe(prompt,
-                 num_inference_steps=int(steps),
-                 guidance_scale=float(guidance),
-                 generator=gen).images[0]
-    return image
+    gen = None if seed == 0 else torch.manual_seed(int(seed))
+    result = pipe(prompt,
+                  num_inference_steps=int(steps),
+                  guidance_scale=float(guidance),
+                  generator=gen)
+    return result.images
 
 with gr.Blocks() as demo:
     gr.Markdown("# Stable Diffusion Text→Image Generation Demo")