Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ download("en_core_web_sm")
|
|
20 |
nlp = spacy.load("en_core_web_sm")
|
21 |
|
22 |
# Load the pre-trained models for image captioning and summarization
|
23 |
-
model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-
|
24 |
model = VisionEncoderDecoderModel.from_pretrained(model_name)
|
25 |
feature_extractor = ViTImageProcessor.from_pretrained(model_name)
|
26 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
20 |
nlp = spacy.load("en_core_web_sm")
|
21 |
|
22 |
# Load the pre-trained models for image captioning and summarization
|
23 |
+
model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-115k-12"
|
24 |
model = VisionEncoderDecoderModel.from_pretrained(model_name)
|
25 |
feature_extractor = ViTImageProcessor.from_pretrained(model_name)
|
26 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|