Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -70,4 +70,4 @@ class PreTrainedPipeline():
|
|
| 70 |
inputs = data#.pop("inputs", data)
|
| 71 |
# process input text
|
| 72 |
prediction = model_infer(self.model, self.tokenizer, inputs+"TL;DR")
|
| 73 |
-
return {"text": prediction[len(inputs
|
|
|
|
| 70 |
inputs = data#.pop("inputs", data)
|
| 71 |
# process input text
|
| 72 |
prediction = model_infer(self.model, self.tokenizer, inputs+"TL;DR")
|
| 73 |
+
return {"text": prediction[len(inputs):]}
|