diff --git a/README.md b/README.md index 38302b0..bced482 100644 --- a/README.md +++ b/README.md @@ -37,8 +37,8 @@ import requests from PIL import Image from transformers import BlipProcessor, BlipForConditionalGeneration -processor = BlipProcessor.from_pretrained("ybelkada/blip-image-captioning-base") -model = BlipForConditionalGeneration.from_pretrained("ybelkada/blip-image-captioning-base") +processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base") +model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base") img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg' raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')