Shriharshan
commited on
Commit
•
99c773c
1
Parent(s):
d837338
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, PreTrainedTokenizerFast
|
2 |
|
3 |
-
model = VisionEncoderDecoderModel.from_pretrained("
|
4 |
vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
|
5 |
tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
|
6 |
|
|
|
1 |
from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, PreTrainedTokenizerFast
|
2 |
|
3 |
+
model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
|
4 |
vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
|
5 |
tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
|
6 |
|