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-09"
+model_name = "NourFakih/Vit-GPT2-COCO2017Flickr-85k-11"
 model = VisionEncoderDecoderModel.from_pretrained(model_name)
 feature_extractor = ViTImageProcessor.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -71,7 +71,7 @@ def search_captions(query, captions):
     
     return results
 
-st.title("Image Gallery with Captioning and Search")
+st.title("Image Captioning Gallery")
 
 # Sidebar for search functionality
 with st.sidebar: