moranyanuka
/

blip-image-captioning-base-mocha

image-text-to-text

Inference Endpoints

Model card Files Files and versions Community

moranyanuka commited on Dec 19, 2023

Commit

576766f

•

1 Parent(s): d38a495

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
-processor = BlipProcessor.from_pretrained("moranyanuka/blip-image-captioning-base-mocha")
 model = BlipForConditionalGeneration.from_pretrained("moranyanuka/blip-image-captioning-base-mocha").to("cuda")
 img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
@@ -88,7 +88,7 @@ import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
-processor = BlipProcessor.from_pretrained("moranyanuka/blip-image-captioning-base-mocha")
 model = BlipForConditionalGeneration.from_pretrained("moranyanuka/blip-image-captioning-base-mocha", torch_dtype=torch.float16).to("cuda")
 img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'

 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 model = BlipForConditionalGeneration.from_pretrained("moranyanuka/blip-image-captioning-base-mocha").to("cuda")
 img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 model = BlipForConditionalGeneration.from_pretrained("moranyanuka/blip-image-captioning-base-mocha", torch_dtype=torch.float16).to("cuda")
 img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'