bug fix

delphi-suite · Feb 16, 2024 · c292da7 · c292da7
1 parent 5af8a6f
commit c292da7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/delphi/eval/token_labelling.py b/src/delphi/eval/token_labelling.py
@@ -165,11 +165,11 @@ def label_batch_sentences(
         corresponding token length where each entry provides the labels/categories
         for the token. Sentence -> Token -> Labels
     """
-    global NLP
+    global NLP, SPACY_MODEL
 
     if NLP is None:
         # Load english language model
-        NLP = spacy.load("en_core_web_trf")
+        NLP = spacy.load(SPACY_MODEL)
     # labelled tokens, list holding sentences holding tokens holding corresponding token labels
     labelled_sentences: list[list[dict[str, bool]]] = list()