JonathanEGP
/

Anonimizador_Ner

Inference Endpoints

Model card Files Files and versions Community

JonathanEGP commited on Jul 23, 2024

Commit

df891cf

·

verified ·

1 Parent(s): 9490472

Update Anonimizador_Ner.py

Files changed (1) hide show

Anonimizador_Ner.py +1 -1

Anonimizador_Ner.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import Pipeline, AutoModelForTokenClassification, AutoTokenize
 class AnonymizationPipeline(Pipeline):
     def __init__(self, model=None, tokenizer=None, **kwargs):
-        self.model_name = "JonathanEGP/Beto_Ner"
         if model is None:
             model = AutoModelForTokenClassification.from_pretrained(self.model_name)

 class AnonymizationPipeline(Pipeline):
     def __init__(self, model=None, tokenizer=None, **kwargs):
+        self.model_name = "JonathanEGP/RoBERTa_base_bn_ner"
         if model is None:
             model = AutoModelForTokenClassification.from_pretrained(self.model_name)