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}
|
|
|
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+"TL;DR"):]}
|