kusumakar commited on
Commit
f9f9974
1 Parent(s): 5149f5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecode
6
  # Directory path to the saved model on Google Drive
7
  model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
8
  feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
9
- tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning
10
 
11
  def generate_captions(image):
12
  image = Image.open(image).convert("RGB")
 
6
  # Directory path to the saved model on Google Drive
7
  model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
8
  feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
9
+ tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
10
 
11
  def generate_captions(image):
12
  image = Image.open(image).convert("RGB")