Update app.py

app.py

@@ -2,12 +2,12 @@ from PIL import Image
 import requests
 import gradio as gr
 
-from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
+from transformers import BlipProcessor, BlipForConditionalGeneration
 
-model_id = "nlpconnect/vit-gpt2-image-captioning"
+model_id = "Salesforce/blip-image-captioning-base"
 
-model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
-processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
+model = BlipForConditionalGeneration.from_pretrained(model_id)
+processor = BlipProcessor.from_pretrained(model_id)
 
 def launch(input):
     image = Image.open(requests.get(input, stream=True).raw).convert('RGB')