Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
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")
|