Hiveurban
/

dictabert-ner-handler

Token Classification

Inference Endpoints

Model card Files Files and versions Community

shamaayan commited on Jan 28, 2024

Commit

8c22397

·

1 Parent(s): 34989a6

EndpointHandler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -3,7 +3,7 @@ from typing import Dict, List, Any
 from tokenizers.decoders import WordPiece
-class EndpointDictaBertNERHandler:
     def __init__(self):
         self.model = pipeline('ner', model='dicta-il/dictabert-ner', aggregation_strategy='simple')
         self.model.tokenizer.backend_tokenizer.decoder = WordPiece()

 from tokenizers.decoders import WordPiece
+class EndpointHandler:
     def __init__(self):
         self.model = pipeline('ner', model='dicta-il/dictabert-ner', aggregation_strategy='simple')
         self.model.tokenizer.backend_tokenizer.decoder = WordPiece()