fix path
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ import requests
|
|
12 |
from PIL import Image
|
13 |
from transformers import ViTFeatureExtractor, AutoTokenizer, FlaxVisionEncoderDecoderModel
|
14 |
|
15 |
-
loc = "ydshieh/
|
16 |
|
17 |
feature_extractor = ViTFeatureExtractor.from_pretrained(loc)
|
18 |
tokenizer = AutoTokenizer.from_pretrained(loc)
|
12 |
from PIL import Image
|
13 |
from transformers import ViTFeatureExtractor, AutoTokenizer, FlaxVisionEncoderDecoderModel
|
14 |
|
15 |
+
loc = "ydshieh/vit-gpt2-coco-en"
|
16 |
|
17 |
feature_extractor = ViTFeatureExtractor.from_pretrained(loc)
|
18 |
tokenizer = AutoTokenizer.from_pretrained(loc)
|