lynn-twinkl
commited on
Commit
·
8b56a5d
1
Parent(s):
893d9c9
Changed default model to sm
Browse files
src/models/topic_modeling_pipeline.py
CHANGED
|
@@ -31,7 +31,7 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
| 31 |
|
| 32 |
## ---------- LOAD SPACY MODEL ---------
|
| 33 |
|
| 34 |
-
def load_spacy_model(model_name="
|
| 35 |
"""
|
| 36 |
This model is used for sentence tokenization
|
| 37 |
as well as stopword generation
|
|
|
|
| 31 |
|
| 32 |
## ---------- LOAD SPACY MODEL ---------
|
| 33 |
|
| 34 |
+
def load_spacy_model(model_name="en_core_web_sm"):
|
| 35 |
"""
|
| 36 |
This model is used for sentence tokenization
|
| 37 |
as well as stopword generation
|