Hugging Face's logo

Spaces:
yashikag
/
image-caption-generator
Running

App Files Community
yashikag commited on
Commit
c4b6d3a
·
1 Parent(s): 04134b2

Update image_caption.py

Browse files
Files changed (1) hide show
  1. image_caption.py +1 -1
image_caption.py CHANGED
@@ -17,7 +17,7 @@ from PIL import Image
17
  from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, PreTrainedTokenizerFast
18
  import requests
19
 
20
- model = VisionEncoderDecoderModel.from_pretrained("sachin/vit2distilgpt2")
21
 
22
  vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
23
 
 
17
  from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, PreTrainedTokenizerFast
18
  import requests
19
 
20
+ model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
21
 
22
  vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
23