Lin0He commited on
Commit
11ee141
·
verified ·
1 Parent(s): c5419b6

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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+"TL;DR"):]}
 
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):]}