harsh13333
/

flair-ner-shipping-labels

Token Classification

sequence-tagger-model

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

harsh13333 commited on Mar 29

Commit

1280f85

•

1 Parent(s): 9695220

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -8,7 +8,7 @@ class EndpointHandler():
         self,
         path: str,
     ):
-        self.tagger = SequenceTagger.load("pytorch_model.bin")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """

         self,
         path: str,
     ):
+        self.tagger = SequenceTagger.load("final-model.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """