Add catch for output_count

handler.py

@@ -76,12 +76,14 @@ class EndpointHandler():
         """
         prompt = data.pop("inputs", None)
         image = data.pop("image", None)
-        output_count = data.pop("output_count", 1)
+        output_count = data.pop("output_count", None)
         controlnet_type = data.pop("controlnet_type", None)
         
         # Check if neither prompt nor image is provided
         if prompt is None and image is None:
             return {"error": "Please provide a prompt and base64 encoded image."}
+        if output_count is None:
+            return {"error": "Please provide an output_count."}
         
         # Check if a new controlnet is provided
         if controlnet_type is not None and controlnet_type != self.control_type: