Linear-Matrix-Probability commited on
Commit
930f282
·
verified ·
1 Parent(s): 6bfc699

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -16,8 +16,8 @@ CONFIG = {
16
 
17
  class EndpointHandler():
18
  def __init__(self,
19
- model_path: str = "zjunlp/chatcell-small",
20
- tokenizer_path: str = "zjunlp/chatcell-small"):
21
 
22
  self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
23
  self.model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 
16
 
17
  class EndpointHandler():
18
  def __init__(self,
19
+ model_path: str = ".",
20
+ tokenizer_path: str = "."):
21
 
22
  self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
23
  self.model = AutoModelForSeq2SeqLM.from_pretrained(model_path)