spottedzebra
/

quantized_onnx_skill_signal

Token Classification

Model card Files Files and versions

shubhasz commited on Sep 7, 2023

Commit

be0ec28

·

1 Parent(s): be80ddf

small bug

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -36,7 +36,7 @@ class EndpointHandler():
     def __init__(self, path="") -> None:
         model = ORTModelForTokenClassification.from_pretrained(path)
         tokenizer = AutoTokenizer.from_pretrained(path)
-        self.pipe = MyTokenClassificationPipeline(model=model,
                                                     framework='pt',
                                                     task='ner',
                                                     tokenizer=tokenizer,

     def __init__(self, path="") -> None:
         model = ORTModelForTokenClassification.from_pretrained(path)
         tokenizer = AutoTokenizer.from_pretrained(path)
+        self.pipeline = MyTokenClassificationPipeline(model=model,
                                                     framework='pt',
                                                     task='ner',
                                                     tokenizer=tokenizer,