Update app.py

app.py

@@ -26,7 +26,7 @@ nltk.download('omw-1.4')
 nlp = spacy.load("en_core_web_sm")
 
 # Load the pre-trained model for image captioning
-model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-85k-11"
+model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-85k-09"
 model = VisionEncoderDecoderModel.from_pretrained(model_name)
 feature_extractor = ViTImageProcessor.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -38,7 +38,6 @@ model.config.eos_token_id = tokenizer.eos_token_id
 model.config.decoder_start_token_id = tokenizer.bos_token_id
 model.config.pad_token_id = tokenizer.pad_token_id
 
-
 def generate_caption(image):
     pixel_values = feature_extractor(images=image, return_tensors="pt").pixel_values
     output_ids = model.generate(pixel_values)