Update app.py

app.py

@@ -2,8 +2,8 @@ import io
 import os
 import streamlit as st
 import requests
+from transformers import pipeline
 from PIL import Image
-from model import get_caption_model, generate_caption
 from googletrans import Translator
 import sqlite3
 
@@ -13,10 +13,15 @@ st.set_page_config(page_title="Image Caption Generator", layout="wide")
 translator = Translator()
 
 @st.cache_resource
-def get_model():
-    return get_caption_model()
+# Image to Text
+def image_to_text(url):
+    # Load a transformer
+    image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+
+    text = image_to_text(url)[0]['generated_text']
+    
+    return text
 
-caption_model = get_model()
 
 # Constants
 SIGNUP_SUCCESS_MSG = "Signup successful! You can now login."
@@ -102,14 +107,15 @@ def translate_caption(caption, target_language="en"):
 
 def predict(cap_col, target_language):
     captions = []
-    pred_caption = generate_caption('tmp.jpg', caption_model)
+    url = 'tmp.jgp'
+    pred_caption = image_to_text(url)
 
     cap_col.markdown('#### Predicted Captions:')
     translated_caption = translate_caption(pred_caption, target_language)
     captions.append(translated_caption)
 
     for _ in range(4):
-        pred_caption = generate_caption('tmp.jpg', caption_model, add_noise=True)
+        pred_caption = image_to_text(url)
         if pred_caption not in captions:
             translated_caption = translate_caption(pred_caption, target_language)
             captions.append(translated_caption)