Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 # image2text
 def img2text(img):
     image_to_text_model = pipeline("image-to-text",
-                                   model="Salesforce/blip-image-captioning-base")
+                                   model="nlpconnect/vit-gpt2-image-captioning")
     text = image_to_text_model(img)[0]["generated_text"]
     return text