blip_image_embeddings

radames commited on Apr 14, 2023

Commit

a888eba

1 Parent(s): 152f178

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -43,6 +43,7 @@ class PreTrainedPipeline():
             A :obj:`dict`:. The object returned should be a dict like {"feature_vector": [0.6331314444541931,0.8802216053009033,...,-0.7866355180740356,]} containing :
                 - "feature_vector": A list of floats corresponding to the image embedding.
         """
         parameters = {"mode": "image"}
         if isinstance(inputs, str):
             # decode base64 image to PIL

             A :obj:`dict`:. The object returned should be a dict like {"feature_vector": [0.6331314444541931,0.8802216053009033,...,-0.7866355180740356,]} containing :
                 - "feature_vector": A list of floats corresponding to the image embedding.
         """
         parameters = {"mode": "image"}
         if isinstance(inputs, str):
             # decode base64 image to PIL