Update handler.py

handler.py

@@ -124,7 +124,6 @@ class EndpointHandler():
             image=control_image,
             num_inference_steps=num_inference_steps,
             guidance_scale=guidance_scale,
-            num_images_per_prompt=4,
             height=height,
             width=width,
             controlnet_conditioning_scale=controlnet_conditioning_scale,
@@ -133,7 +132,7 @@ class EndpointHandler():
 
         # generator=self.generator COMMENTED from self.pipe
         # return the first generated PIL image
-        return out.images
+        return out.images[0]
 
     # helper to decode input image
     def decode_base64_image(self, image_string):