Update app.py

app.py

@@ -5,22 +5,19 @@ import requests
 from PIL import Image
 from model import get_caption_model, generate_caption
 
-# Use st.experimental_memo instead of st.cache_resource
-@st.experimental_memo(allow_output_mutation=True)
-def get_model():
-    return get_caption_model()
-
-caption_model = get_model()
+# Initialize session state for the model
+if 'caption_model' not in st.session_state:
+    st.session_state.caption_model = get_caption_model()
 
 def predict():
     captions = []
     try:
-        pred_caption = generate_caption('tmp.jpg', caption_model)
+        pred_caption = generate_caption('tmp.jpg', st.session_state.caption_model)
         st.markdown('#### Predicted Captions:')
         captions.append(pred_caption)
 
         for _ in range(4):
-            pred_caption = generate_caption('tmp.jpg', caption_model, add_noise=True)
+            pred_caption = generate_caption('tmp.jpg', st.session_state.caption_model, add_noise=True)
             if pred_caption not in captions:
                 captions.append(pred_caption)