NourFakih commited on
Commit
2cf13ae
·
verified ·
1 Parent(s): 69c36e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ nltk.download('omw-1.4')
26
  nlp = spacy.load("en_core_web_sm")
27
 
28
  # Load the pre-trained model for image captioning
29
- model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-85k-09"
30
  model = VisionEncoderDecoderModel.from_pretrained(model_name)
31
  feature_extractor = ViTImageProcessor.from_pretrained(model_name)
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
26
  nlp = spacy.load("en_core_web_sm")
27
 
28
  # Load the pre-trained model for image captioning
29
+ model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-85k-11"
30
  model = VisionEncoderDecoderModel.from_pretrained(model_name)
31
  feature_extractor = ViTImageProcessor.from_pretrained(model_name)
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)