Spaces:
Sleeping
Sleeping
Update apps/intro.py
Browse files- apps/intro.py +2 -1
apps/intro.py
CHANGED
|
@@ -13,6 +13,7 @@ def init_models():
|
|
| 13 |
REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
|
| 14 |
FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
|
| 15 |
umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
|
|
|
|
| 16 |
|
| 17 |
def app():
|
| 18 |
word_to_embed_list = st.session_state['embed_list']
|
|
@@ -29,7 +30,7 @@ def app():
|
|
| 29 |
if st.button("Embed"):
|
| 30 |
with st.spinner("👑 Embedding your input"):
|
| 31 |
|
| 32 |
-
init_models()
|
| 33 |
|
| 34 |
|
| 35 |
word_to_embed_list.append(word_to_embed)
|
|
|
|
| 13 |
REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
|
| 14 |
FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
|
| 15 |
umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
|
| 16 |
+
return model, umap_model
|
| 17 |
|
| 18 |
def app():
|
| 19 |
word_to_embed_list = st.session_state['embed_list']
|
|
|
|
| 30 |
if st.button("Embed"):
|
| 31 |
with st.spinner("👑 Embedding your input"):
|
| 32 |
|
| 33 |
+
model, umap_model = init_models()
|
| 34 |
|
| 35 |
|
| 36 |
word_to_embed_list.append(word_to_embed)
|