Update apps/intro.py

apps/intro.py

@@ -7,6 +7,13 @@ import pandas as pd
 import os
 import joblib
 
+def init_models():
+    model_name = 'sentence-transformers/all-MiniLM-L6-v2'
+    model = SentenceTransformer(model_name)
+    REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
+    FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
+    umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
+
 def app():
     word_to_embed_list =  st.session_state['embed_list']
     cat_list =  st.session_state['cat_list']
@@ -20,12 +27,10 @@ def app():
 
         
         if st.button("Embed"):
-            with st.spinner("👑 load language model (sentence transformer)"):
-                model_name = 'sentence-transformers/all-MiniLM-L6-v2'
-                model = SentenceTransformer(model_name)
-                REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
-                FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
-                umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
+            with st.spinner("👑 Embedding your input"):
+                
+                init_models()
+                
                 
                 word_to_embed_list.append(word_to_embed)
                 st.session_state['embed_list'] = word_to_embed_list