Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -80,5 +80,5 @@ class EndpointHandler():
|
|
| 80 |
# process input
|
| 81 |
inputs = data.pop("inputs", data)
|
| 82 |
# process input text
|
| 83 |
-
prediction = model_infer( self.model, self.tokenizer
|
| 84 |
return prediction
|
|
|
|
| 80 |
# process input
|
| 81 |
inputs = data.pop("inputs", data)
|
| 82 |
# process input text
|
| 83 |
+
prediction = model_infer( self.model, self.tokenizer, inputs+"TL;DR")
|
| 84 |
return prediction
|