Hugging Face's logo

CiaranMn
/
blip-captioning

Image-to-Text
generic
image-captioning
endpoints-template
Inference Endpoints
Model card Files Community
CiaranMn commited on
Commit
5e340bc
•
1 Parent(s): b3f0971

missing bracket

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -35,7 +35,7 @@ class EndpointHandler():
35
  inputs = data.pop("inputs", data)
36
  parameters = data.pop("parameters", {})
37
 
38
- raw_image = Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
39
 
40
  processed_image = self.processor(images=raw_images, return_tensors="pt")
41
  processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
 
35
  inputs = data.pop("inputs", data)
36
  parameters = data.pop("parameters", {})
37
 
38
+ raw_image = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
39
 
40
  processed_image = self.processor(images=raw_images, return_tensors="pt")
41
  processed_image["pixel_values"] = processed_image["pixel_values"].to(device)