jla25 commited on
Commit
25fd307
·
verified ·
1 Parent(s): c1975d1

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -4
handler.py CHANGED
@@ -5,14 +5,14 @@ import json
5
 
6
  model_name = "jla25/squareV3"
7
 
8
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
9
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=True)
10
 
11
 
12
  class EndpointHandler:
13
  def __init__(self, model_dir):
14
- self.tokenizer = AutoTokenizer.from_pretrained(model_dir, use_auth_token=True)
15
- self.model = AutoModelForSeq2SeqLM.from_pretrained(model_dir, use_auth_token=True)
16
  self.model.eval()
17
 
18
  def preprocess(self, data):
 
5
 
6
  model_name = "jla25/squareV3"
7
 
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
9
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10
 
11
 
12
  class EndpointHandler:
13
  def __init__(self, model_dir):
14
+ self.tokenizer = AutoTokenizer.from_pretrained(model_dir)
15
+ self.model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)
16
  self.model.eval()
17
 
18
  def preprocess(self, data):