Update app.py

app.py

@@ -39,7 +39,7 @@ def infer(
     else:
         generator = g_cpu.manual_seed(seed)
 
-    output_images = pipe(
+    output_image = pipe(
         prompt,
         conditioning_image,
         height=size,
@@ -49,9 +49,9 @@ def infer(
         negative_prompt=negative_prompt,
         guidance_scale=guidance_scale,
         controlnet_conditioning_scale=1.0,
-    ).images
+    ).images[0]
 
-    return output_images
+    return output_image
 
 with gr.Blocks() as demo:
     gr.Markdown(