cguynup commited on
Commit
bba26fd
·
1 Parent(s): ed8a7fb

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -2
handler.py CHANGED
@@ -7,11 +7,10 @@ class EndpointHandler():
7
  def __init__(self, path=""):
8
  # load the optimized model
9
  self.model = ORTModelForSequenceClassification.from_pretrained(path)
10
- self.tokenizer = AutoTokenizer.from_pretrained(path)
11
 
12
 
13
  def __call__(self, data):
14
-
15
  answers = data.pop("answers")
16
  paraphrases = data.pop("paraphrases")
17
 
 
7
  def __init__(self, path=""):
8
  # load the optimized model
9
  self.model = ORTModelForSequenceClassification.from_pretrained(path)
10
+ self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', do_lower_case=True)
11
 
12
 
13
  def __call__(self, data):
 
14
  answers = data.pop("answers")
15
  paraphrases = data.pop("paraphrases")
16