Merge remote-tracking branch 'origin' into pseudonymization-tests

ssciwr · Oct 29, 2024 · 7a9defb · 7a9defb
2 parents bb07b5e + f8e9afd
commit 7a9defb
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mailcom/parse.py b/mailcom/parse.py
@@ -86,13 +86,13 @@ def init_spacy(self, language: str, model="default"):
     def init_transformers(
         self,
         model="xlm-roberta-large-finetuned-conll03-english",
-        model_revision_number="default",
+        model_revision_number="18f95e9",
     ):
-        # TODO: Model revision number
-
-        # ner_recognizer = pipeline("token-classification")
         self.ner_recognizer = pipeline(
-            "token-classification", model=model, aggregation_strategy="simple"
+            "token-classification",
+            model=model,
+            revision=model_revision_number,
+            aggregation_strategy="simple",
         )
 
     def reset(self):