update: update return type of inference

handler.py

@@ -152,7 +152,7 @@ class EndpointHandler():
             image, width=1024, height=1024)
 
         # run inference pipeline
-        images = self.pipe(
+        out = self.pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
             image=control_image,
@@ -163,9 +163,9 @@ class EndpointHandler():
             num_images_per_prompt=num_of_images,
             controlnet_conditioning_scale=controlnet_conditioning_scale,
             generator=self.generator
-        ).images[0]
+        )
 
-        return images
+        return out.images
 
     # helper to decode input image
     def decode_base64_image(self, image_string):