ben-xl8
/

wmt22-cometkiwi-da

Inference Endpoints

Model card Files Files and versions Community

Ben Nguyen commited on Jul 11, 2023

Commit

b1cf76b

•

1 Parent(s): 7ffa8af

Change ckpt path

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -4,7 +4,7 @@ from comet import load_from_checkpoint
 class EndpointHandler():
     def __init__(self, path=""):
-        self.model = load_from_checkpoint("checkpoints/model.ckpt")
     def __call__(self, data: Dict[str, Any]) -> List[Any]:
         """

 class EndpointHandler():
     def __init__(self, path=""):
+        self.model = load_from_checkpoint("./checkpoints/model.ckpt")
     def __call__(self, data: Dict[str, Any]) -> List[Any]:
         """