fix

handler.py

@@ -21,15 +21,20 @@ class EndpointHandler:
         This method is called for every API request.
         
         Args:
-            data (Dict): The input data dictionary. Expects "inputs" key with image bytes.
+            data (Dict): The input data dictionary. Can be PIL Image or bytes.
         
         Returns:
             Dict[str, str]: A dictionary with a single key "generated_text", 
                             containing a JSON string of the results.
         """
-        # Get image bytes from the request
+        # Get image from the request
         inputs = data.pop("inputs", data)
-        image = Image.open(io.BytesIO(inputs))
+        
+        # Handle both PIL Image objects (from image content-type) and bytes (from JSON)
+        if isinstance(inputs, Image.Image):
+            image = inputs
+        else:
+            image = Image.open(io.BytesIO(inputs))
 
         # Pass the image to the pipeline
         prediction = self.pipe(image)