radames
/

blip_image_embeddings

Feature Extraction

Inference Endpoints

Model card Files Files and versions Community

radames commited on Apr 13, 2023

Commit

bfe7523

•

1 Parent(s): c0ca149

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -48,8 +48,8 @@ class PreTrainedPipeline():
             # decode base64 image to PIL
             image = Image.open(
                 BytesIO(base64.b64decode(inputs))).convert("RGB")
-        elif isinstance(inputs, Image.Image):
-            image = inputs.convert("RGB")
         image = self.transform(image).unsqueeze(0).to(device)

             # decode base64 image to PIL
             image = Image.open(
                 BytesIO(base64.b64decode(inputs))).convert("RGB")
+        elif isinstance(inputs, "Image.Image"):
+            image = Image.open(inputs).convert("RGB")
         image = self.transform(image).unsqueeze(0).to(device)