used caching to improve speed when switching tabs

app.py

@@ -12,24 +12,39 @@ from streamlit_tags import st_tags, st_tags_sidebar
 
 st.set_page_config(page_title="Ancient Greek Word2Vec", layout="centered")
 
+# Cache data
+@st.cache_data
+def load_lsj_dict():
+    return json.load(open('lsj_dict.json', 'r'))
+
+@st.cache_data
+def load_all_models_words():
+    return load_compressed_word_list('all_lemmas.pkl.gz')
+
+
+# Load compressed word list
+all_models_words = load_all_models_words()
+
+
+# Prepare lsj dictionary
+lemma_dict = load_lsj_dict()
+
+
 # Horizontal menu
 active_tab = option_menu(None, ["Nearest neighbours", "Cosine similarity", "3D graph", 'Dictionary'], 
     menu_icon="cast", default_index=0, orientation="horizontal")
 
-# Prepare dictionary
-lemma_dict = json.load(open('lsj_dict.json', 'r'))
 
 # Nearest neighbours tab
 if active_tab == "Nearest neighbours":
     
     # Load the compressed word list
-    compressed_word_list_filename = 'corpora/compass_filtered.pkl.gz'
-    all_words = load_compressed_word_list(compressed_word_list_filename)
+
     eligible_models = ["Archaic", "Classical", "Hellenistic", "Early Roman", "Late Roman"]
     
     with st.form("nn_form"):
         st.markdown("## Nearest Neighbours")
-        target_word = st.multiselect("Enter a word", all_words, max_selections=1)
+        target_word = st.multiselect("Enter a word", all_models_words, max_selections=1)
         if len(target_word) > 0:
             target_word = target_word[0]
             
@@ -113,12 +128,11 @@ elif active_tab == "3D graph":
     col1, col2 = st.columns(2)
     
     # Load compressed word list
-    compressed_word_list_filename = 'corpora/compass_filtered.pkl.gz'
-    all_words = load_compressed_word_list(compressed_word_list_filename)
+    all_models_words = load_all_models_words()
     
     with st.container():
         with col1:
-            word = st.multiselect("Enter a word", all_words, max_selections=1)
+            word = st.multiselect("Enter a word", all_models_words, max_selections=1)
             if len(word) > 0:
                 word = word[0]