Upload handler.py

added text input and single image

handler.py

@@ -24,14 +24,12 @@ class EndpointHandler():
         """
 
         inputs = data.pop("inputs", data)
+        text = data.get("text", "")
         parameters = data.pop("parameters", {})
 
-        if isinstance(inputs, bytes):  # If a single image is passed
-            raw_images = [Image.open(BytesIO(inputs))]
-        else:  # If already a list of images
-            raw_images = [Image.open(BytesIO(_img)) for _img in inputs]
+        raw_images = Image.open(BytesIO(inputs)).convert("")
 
-        processed_image = self.processor(images=raw_images, return_tensors="pt")
+        processed_image = self.processor(images=raw_images, text = text, return_tensors="pt")
         processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
         processed_image = {**processed_image, **parameters}