paragon-AI
/

blip2-image-to-text

visual-question-answering

image-captioning

Inference Endpoints

Model card Files Files and versions Community

thoth-AI commited on Jun 24, 2023

Commit

f30af1b

·

1 Parent(s): f49e3ec

Updated handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -31,7 +31,7 @@ class EndpointHandler:
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
-        raw_images = inputs
         inputs = self.processor(inputs, return_tensors="pt").to("cuda")

         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
+        raw_images = Image.open(BytesIO(inputs))
         inputs = self.processor(inputs, return_tensors="pt").to("cuda")