quadranttechnologies
/

qhub-blip-image-captioning-finetuned

image-text-to-text

Inference Endpoints

Model card Files Files and versions Community

sguna commited on 8 days ago

Commit

f83f8ef

•

1 Parent(s): 138ad6d

Upload handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -30,7 +30,7 @@ class EndpointHandler():
         text = data.get("text", "")
         parameters = data.pop("parameters", {})
-        raw_images = Image.open(BytesIO(base64.b64decode(inputs))).convert("RGB")
         processed_image = self.processor(images=raw_images, text=text, return_tensors="pt")
         processed_image["pixel_values"] = processed_image["pixel_values"].to(device)

         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["pixel_values"] = processed_image["pixel_values"].to(device)