Update app.py

app.py

@@ -5,7 +5,7 @@ import streamlit as st
 # function part
 # img2text
 def img2text(url):
-    image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")  # model is slow
+    image_to_text_model = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
     text = image_to_text_model(url)[0]["generated_text"]
     return text