From 5af8a6f6d22c1fdfb0962ea0aa71efbc21195276 Mon Sep 17 00:00:00 2001
From: Joshua Wendland <80349780+joshuawe@users.noreply.github.com>
Date: Fri, 16 Feb 2024 11:52:44 +0100
Subject: [PATCH] change spacy model from "trf" to "sm"

---
 src/delphi/eval/token_labelling.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/delphi/eval/token_labelling.py b/src/delphi/eval/token_labelling.py
index 55cecc02..d2d2b45e 100644
--- a/src/delphi/eval/token_labelling.py
+++ b/src/delphi/eval/token_labelling.py
@@ -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] = {