Update app.py

app.py

@@ -123,7 +123,7 @@ def infer(
     
 
     # Generate the image
-    image = pipe(
+    images = pipe(
         image=pil_images if len(pil_images) > 0 else None,
         prompt=prompt,
         height=height,
@@ -136,7 +136,7 @@ def infer(
     ).images
 
     # Return images, seed, and make button visible
-    return image, seed, gr.update(visible=True)
+    return images[0], seed, gr.update(visible=True)
 
 # --- Examples and UI Layout ---
 examples = []