BashirRP
/

llm_judge2

Inference Endpoints

Model card Files Files and versions Community

Bashir Rastegarpanah commited on Jan 10

Commit

c710467

•

1 Parent(s): 9868e73

add custom handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -11,7 +11,7 @@ class EndpointHandler:
         # pseudo:
         # self.model= load_model(path)
         self.model = model
-        self.tokenizer = AutoTokenizer.from_pretrained("roberta-large", padding_side=padding_side)
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

         # pseudo:
         # self.model= load_model(path)
         self.model = model
+        self.tokenizer = AutoTokenizer.from_pretrained("roberta-large", padding_side='right')
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: