Update README.md
Browse files
README.md
CHANGED
@@ -11,8 +11,8 @@ license: bsd-3-clause
|
|
11 |
import requests
|
12 |
from PIL import Image
|
13 |
from transformers import BlipProcessor, BlipForConditionalGeneration
|
14 |
-
processor = BlipProcessor.from_pretrained("
|
15 |
-
model = BlipForConditionalGeneration.from_pretrained("
|
16 |
img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
|
17 |
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|
18 |
# conditional image captioning
|
|
|
11 |
import requests
|
12 |
from PIL import Image
|
13 |
from transformers import BlipProcessor, BlipForConditionalGeneration
|
14 |
+
processor = BlipProcessor.from_pretrained("Revrse/icon-captioning-model")
|
15 |
+
model = BlipForConditionalGeneration.from_pretrained("Revrse/icon-captioning-model")
|
16 |
img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
|
17 |
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|
18 |
# conditional image captioning
|