diff --git a/mailcom/parse.py b/mailcom/parse.py index 81ea972..e0579c6 100644 --- 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):