luodian commited on
Commit
35d9588
1 Parent(s): 60667f7

Update README.md

Browse files

There's a type in loading BLIPConditionalGeneration model.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -73,7 +73,7 @@ from PIL import Image
73
  from transformers import BlipProcessor, BlipForConditionalGeneration
74
 
75
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
76
- model = BlipForConditionalGeneration.from_pretrained("Salesfoce/blip-image-captioning-base").to("cuda")
77
 
78
  img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
79
  raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
 
73
  from transformers import BlipProcessor, BlipForConditionalGeneration
74
 
75
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
76
+ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to("cuda")
77
 
78
  img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
79
  raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')