Upload handler.py

handler.py

@@ -26,13 +26,11 @@ class EndpointHandler():
                 - "description": A string corresponding to the generated description.
         """
 
-        inputs = data.pop("inputs", data)
+        images = data.pop("inputs", data)
         text = data.get("text", "")
         parameters = data.pop("parameters", {})
 
-        raw_images = Image.open(inputs).convert("RGB")
-
-        processed_image = self.processor(images=raw_images, text=text, return_tensors="pt")
+        processed_image = self.processor(images=images, text=text, return_tensors="pt")
         processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
         processed_image = {**processed_image, **parameters}