Update handler.py

handler.py

@@ -7,14 +7,14 @@ class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         self.model = ORTModelForSequenceClassification.from_pretrained(path)
-        self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', do_lower_case=True)
+        self.tokenizer = AutoTokenizer.from_pretrained(path)
 
 
     def __call__(self, data):
-        answers = data.pop("answers")
-        paraphrases = data.pop("paraphrases")
+        answers = data.pop("inputs")
+        #paraphrases = data.pop("paraphrases")
 
-        inputs = self.tokenizer(answers, paraphrases, max_length=253, padding=True, truncation=True, return_tensors='pt')
+        inputs = self.tokenizer(answers, max_length=253, padding=True, truncation=True, return_tensors='pt')
 
         with torch.no_grad():
             outputs = self.model(**inputs)