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()
|