Add check for count

handler.py

@@ -76,12 +76,15 @@ class EndpointHandler():
         """
         prompt = data.pop("inputs", None)
         image = data.pop("image", None)
-        count = data.pop("count", 1)
+        count = data.pop("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 count is None:
+            return {"error": "Please provide a count."}
         
         # Check if a new controlnet is provided
         if controlnet_type is not None and controlnet_type != self.control_type: