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="en_core_web_md"):
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