Update app.py

app.py

@@ -4,10 +4,10 @@ from transformers import pipeline
 from gtts import gTTS
 import os
 
-# Function to convert image to text using Hugging Face's BLIP model
+# Function to convert image to text using Hugging Face's vit-gpt2-image-captioning model
 def img2text(url):
     """
-    Converts an image to text using the Salesforce/blip-image-captioning-base model.
+    Converts an image to text using the nlpconnect/vit-gpt2-image-captioning model.
     
     Args:
         url (str): Path to the image file.
@@ -15,7 +15,7 @@ def img2text(url):
     Returns:
         str: Generated text caption from the image, without words like "illustration".
     """
-    image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+    image_to_text_model = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
     text = image_to_text_model(url)[0]["generated_text"]
     
     # Remove unwanted words like "illustration"