Update app.py

app.py

@@ -124,7 +124,7 @@ def generate(
         negative_prompt_2 = None  # type: ignore
 
     if use_controlnet:
-        image = pipe(
+        images = pipe(
             prompt=prompt,
             image=image,
             control_image=canny_image,
@@ -137,7 +137,7 @@ def generate(
             num_inference_steps=num_inference_steps_base,
             generator=generator,
         ).images[0]
-        return image
+        return images
 
 with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
     gr.HTML(