Update app.py

app.py

@@ -61,10 +61,19 @@ def main():
     Explore how different words and phrases relate to each other in the embedding space!
     """)
 
+    # Load model at the beginning
+    model_choice = "BERT"  # Default model
+    tokenizer, model = load_model(model_choice)
+
+    
     # Sidebar
     with st.sidebar:
         st.header("Controls")
         model_choice = st.selectbox("Choose a model:", ["BERT", "RoBERTa"])
+
+        if model_choice != st.session_state.get('last_model_choice'):
+            tokenizer, model = load_model(model_choice)
+            st.session_state.last_model_choice = model_choice
         
         new_phrase = st.text_input("Enter a new word or phrase:", "")
         if st.button("Add Phrase"):