cguynup commited on
Commit
f8b5e2b
·
1 Parent(s): 682b1c5

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -11,10 +11,10 @@ class EndpointHandler():
11
 
12
 
13
  def __call__(self, data):
14
- answers = data.pop("inputs")
15
- #paraphrases = data.pop("paraphrases")
16
 
17
- inputs = self.tokenizer(answers, max_length=253, padding=True, truncation=True, return_tensors='pt')
18
 
19
  with torch.no_grad():
20
  outputs = self.model(**inputs)
 
11
 
12
 
13
  def __call__(self, data):
14
+ answers = data.pop("answers")
15
+ paraphrases = data.pop("paraphrases")
16
 
17
+ inputs = self.tokenizer(answers, paraphrases, max_length=253, padding=True, truncation=True, return_tensors='pt')
18
 
19
  with torch.no_grad():
20
  outputs = self.model(**inputs)