Input image as image

handler.py

@@ -39,10 +39,10 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
 
-        raw_images = [Image.open(io.BytesIO(_img)) for _img in inputs]
+        img = Image.open(io.BytesIO(inputs))
 
         # img = Image.open(io.BytesIO(image_bytes)).convert("RGB")
-        img = raw_images[0]
+        # img = raw_images[0]
         
         # SAM requires input prompts, so we'll generate a center point prompt
         height, width = img.size[1], img.size[0]  # PIL returns (width, height)