NourFakih commited on
Commit
7f5d162
·
verified ·
1 Parent(s): ad3be12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-85k-09"
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)