Commit
·
d1b6429
1
Parent(s):
b1a4162
Update README.md
Browse files
README.md
CHANGED
|
@@ -37,8 +37,8 @@ import requests
|
|
| 37 |
from PIL import Image
|
| 38 |
from transformers import BlipProcessor, BlipForConditionalGeneration
|
| 39 |
|
| 40 |
-
processor = BlipProcessor.from_pretrained("
|
| 41 |
-
model = BlipForConditionalGeneration.from_pretrained("
|
| 42 |
|
| 43 |
img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
|
| 44 |
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|
|
|
|
| 37 |
from PIL import Image
|
| 38 |
from transformers import BlipProcessor, BlipForConditionalGeneration
|
| 39 |
|
| 40 |
+
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
| 41 |
+
model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
| 42 |
|
| 43 |
img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
|
| 44 |
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|