Skip to content

Commit

Permalink
change spacy model from "trf" to "sm"
Browse files Browse the repository at this point in the history
  • Loading branch information
joshuawe committed Feb 16, 2024
1 parent ef0f2e4 commit 5af8a6f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/delphi/eval/token_labelling.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@
# make sure the english language model capabilities are installed by the equivalent of:
# python -m spacy download en_core_web_sm
# Should be run once, initially. Download only starts if not already installed.
SPACY_MODEL = "en_core_web_trf"
SPACY_MODEL = "en_core_web_sm" # small: "en_core_web_sm", large: "en_core_web_trf"
NLP = None # global var to hold the language model
if not is_package(SPACY_MODEL):
spacy.cli.download("en_core_web_trf", False, False)
spacy.cli.download(SPACY_MODEL, False, False)


TOKEN_LABELS: dict[str, Callable] = {
Expand Down

0 comments on commit 5af8a6f

Please sign in to comment.