kusumakar commited on
Commit
15511a5
1 Parent(s): 1775378

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from nltk.corpus import stopwords
12
  from nltk.tokenize import word_tokenize
13
  from sklearn.feature_extraction.text import TfidfVectorizer
14
  from sklearn.metrics.pairwise import cosine_similarity
15
- from transformers import AutoTokenizer, VisionEncoderDecoderModel, ViTFeatureExtractor
16
 
17
  # Directory path to the saved model on Google Drive
18
  model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 
12
  from nltk.tokenize import word_tokenize
13
  from sklearn.feature_extraction.text import TfidfVectorizer
14
  from sklearn.metrics.pairwise import cosine_similarity
15
+ from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
16
 
17
  # Directory path to the saved model on Google Drive
18
  model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")