zjunlp
/

chatcell-small

Text Generation

text2text-generation

single-cell analysis

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Linear-Matrix-Probability commited on Feb 14

Commit

2e98537

•

1 Parent(s): 930f282

Update handler.py

Files changed (1) hide show

handler.py +4 -6

handler.py CHANGED Viewed

@@ -14,13 +14,11 @@ CONFIG = {
     'do_sample': True,
 }
-class EndpointHandler():
-    def __init__(self,
-                 model_path: str = ".",
-                 tokenizer_path: str = "."):
-        self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
-        self.model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
     def __call__(self, data: Dict[str, str]) -> Dict[str, str]:

     'do_sample': True,
 }
+class EndpointHandler:
+    def __init__(self, path: str = ""):
+        self.tokenizer = AutoTokenizer.from_pretrained(path)
+        self.model = AutoModelForSeq2SeqLM.from_pretrained(path)
     def __call__(self, data: Dict[str, str]) -> Dict[str, str]: