Salesforce
/
blip-image-captioning-large
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md (#4) 

- Update README.md (18221a40e4da5b2ff3f880de931b7ac631e39935)


Co-authored-by: Daniele Acquaviva <dacquaviva@users.noreply.huggingface.co>
This commit is contained in:
Younes Belkada 2023-01-20 17:00:33 +00:00 committed by system
parent 0b7cde49b7
commit 18903c0832
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

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