Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -24,7 +24,7 @@ class EndpointHandler():
|
|
| 24 |
"""
|
| 25 |
# get inputs
|
| 26 |
text = data.pop("text", data)
|
| 27 |
-
tokens = self.tokenizer(text, return_tensors='pt', padding=True).to(self.device)
|
| 28 |
outputs = self.model(**tokens)
|
| 29 |
results = []
|
| 30 |
for idx, x in enumerate(outputs.logits):
|
|
|
|
| 24 |
"""
|
| 25 |
# get inputs
|
| 26 |
text = data.pop("text", data)
|
| 27 |
+
tokens = self.tokenizer(text, return_tensors='pt', padding=True, truncation=True).to(self.device)
|
| 28 |
outputs = self.model(**tokens)
|
| 29 |
results = []
|
| 30 |
for idx, x in enumerate(outputs.logits):
|