Revert output count

handler.py

@@ -99,7 +99,6 @@ class EndpointHandler():
         height = data.pop("height", None)
         width = data.pop("width", None)
         controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
-        output_count = data.pop("output_count", None)
         
         # process image
         image = self.decode_base64_image(image)
@@ -121,7 +120,7 @@ class EndpointHandler():
 
         
         # return first generate PIL image
-        return out.images[0, output_count]
+        return out.images[0]
     
     # helper to decode input image
     def decode_base64_image(self, image_string):