Salesforce
/

blip-image-captioning-base

luodian commited on Apr 4, 2023

Commit

62ef7fd

•

1 Parent(s): 60667f7

Update README.md (#11)

- Update README.md (35d95889551a32b526b658aefa7842de6ad6eec1)

Co-authored-by: Bo Li <luodian@users.noreply.huggingface.co>

Files changed (1) hide show

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
 processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-model = BlipForConditionalGeneration.from_pretrained("Salesfoce/blip-image-captioning-base").to("cuda")
 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')

 from transformers import BlipProcessor, BlipForConditionalGeneration
 processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to("cuda")
 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')