Update app.py

app.py

@@ -7,11 +7,11 @@ import csv
 from PIL import Image
 
 device = 'cpu'
-
+model_name="NourFakih/Vit-GPT2-COCO2017Flickr-40k-04"
 # Load the pretrained model, feature extractor, and tokenizer
-model = VisionEncoderDecoderModel.from_pretrained("NourFakih/Vit-GPT2-COCO2017Flickr-02").to(device)
-feature_extractor = ViTImageProcessor.from_pretrained("NourFakih/Vit-GPT2-COCO2017Flickr-02")
-tokenizer = AutoTokenizer.from_pretrained("NourFakih/Vit-GPT2-COCO2017Flickr-02")
+model = VisionEncoderDecoderModel.from_pretrained(model_name).to(device)
+feature_extractor = ViTImageProcessor.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 def predict(image, max_length=64, num_beams=4):
     # Process the input image