download spacy model to support all models

getzep · Jan 24, 2024 · 77d80ad · 77d80ad
1 parent edab92e
commit 77d80ad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/app/entity_extractor.py b/app/entity_extractor.py
@@ -89,6 +89,7 @@ def get_extractor() -> SpacyExtractor:
         f"Loading spacy model {settings.nlp_spacy_model}. If the model is not already"
         " downloaded, this may take a while."
     )
+    spacy.cli.download(settings.nlp_spacy_model)
     nlp = spacy.load(settings.nlp_spacy_model)
     log.info(f"Loaded spacy model {settings.nlp_spacy_model}.")