stefanoviel
commited on
Commit
·
53b63ed
1
Parent(s):
e0405d0
forcing download
Browse files- src/streamlit_app.py +3 -3
src/streamlit_app.py
CHANGED
@@ -16,12 +16,12 @@ DATA_FILE = '/tmp/papers_data.pkl'
|
|
16 |
CSV_FILE = 'papers_with_abstracts_parallel.csv'
|
17 |
|
18 |
# --- Caching Functions ---
|
19 |
-
|
20 |
def load_embedding_model():
|
21 |
"""Loads the Sentence Transformer model and caches it."""
|
22 |
-
return SentenceTransformer(EMBEDDING_MODEL)
|
|
|
23 |
|
24 |
-
@st.cache_resource
|
25 |
def load_spell_checker():
|
26 |
"""Loads the SpellChecker object and caches it."""
|
27 |
return SpellChecker()
|
|
|
16 |
CSV_FILE = 'papers_with_abstracts_parallel.csv'
|
17 |
|
18 |
# --- Caching Functions ---
|
19 |
+
|
20 |
def load_embedding_model():
|
21 |
"""Loads the Sentence Transformer model and caches it."""
|
22 |
+
return SentenceTransformer(EMBEDDING_MODEL, force_download=True)
|
23 |
+
|
24 |
|
|
|
25 |
def load_spell_checker():
|
26 |
"""Loads the SpellChecker object and caches it."""
|
27 |
return SpellChecker()
|