Update model.py

model.py

@@ -124,7 +124,7 @@ class Model:
             num_inference_steps=num_steps,
             generator=generator,
             image=control_image,
-        ).images[0]
+        ).images
 
     
     def resize_image(self, image):
@@ -173,7 +173,7 @@ class Model:
         results = self.run_pipe(
             prompt=prompt, negative_prompt=negative_prompt, control_image=canny_image, num_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale), seed=seed, num_images=1,
         )
-        return [canny_image] + results
+        return [canny_image, results[0]]