22

app.py

@@ -140,7 +140,7 @@ def generate_image(
 
     generator = torch.Generator().manual_seed(seed)
 
-    image = pipe_imagen(
+    output_image = pipe_imagen(
         prompt=prompt,
         negative_prompt=negative_prompt,
         num_inference_steps=num_inference_steps,
@@ -149,9 +149,9 @@ def generate_image(
         # generator=generator,
         # true_cfg_scale=guidance_scale,
         # guidance_scale=1.0  # Use a fixed default for distilled guidance
-    ).images[0]
+    )
 
-    return image, seed
+    return output_image, seed
 
     return True