philschmid
/

layoutlm-funsd

Other PyTorch TensorBoard

generic layoutlm generated_from_trainer endpoints-template Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

philschmid HF staff commited on Oct 5, 2022

Commit

1e71dbb

•

1 Parent(s): 3b6df47

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -24,8 +24,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
-        self.model = LayoutLMForTokenClassification.from_pretrained("philschmid/layoutlm-funsd").to(device)
-        self.processor = LayoutLMv2Processor.from_pretrained("philschmid/layoutlm-funsd")
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, List[Any]]:
         """

 class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
+        self.model = LayoutLMForTokenClassification.from_pretrained(path).to(device)
+        self.processor = LayoutLMv2Processor.from_pretrained(path)
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, List[Any]]:
         """