Update README.md
Browse filesThere's a type in loading BLIPConditionalGeneration model.
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("
|
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')
|