Hugging Face's logo

Spaces:
Shriharshan
/
Image-Caption-Generator
Running

App Files Community
Shriharshan commited on
Commit
99c773c
•
1 Parent(s): d837338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, PreTrainedTokenizerFast
2
 
3
- model = VisionEncoderDecoderModel.from_pretrained("Salesforce/blip-image-captioning-large")
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