Update app.py

app.py

@@ -3,18 +3,23 @@ from annotated_text import annotated_text
 from refined.inference.processor import Refined
 
 # Sidebar
-st.sidebar.image("logo-wordlift.png")
+# st.sidebar.image("logo-wordlift.png")
 
 # Initiate the model
 model_options = {"aida_model", "wikipedia_model_with_numbers"}
-selected_model = st.sidebar.selectbox("Select the Model", list(model_options))
+selected_model_name = st.sidebar.selectbox("Select the Model", list(model_options))
 
-# Load the pretrained model
-refined_model = Refined.from_pretrained(model_name=selected_model, entity_set="wikipedia")
+@st.cache_resource  # 👈 Add the caching decorator
+def load_model(model_name):
+    # Load the pretrained model
+    refined_model = Refined.from_pretrained(model_name=model_name, entity_set="wikipedia")
+    return refined_model
+
+# Use the cache model
+refined_model = load_model(selected_model_name)
 
 # Create the form
 with st.form(key='my_form'):
-    st.sidebar.image(image, caption='', use_column_width=True)
     text_input = st.text_input(label='Enter a sentence')
     submit_button = st.form_submit_button(label='Submit')