Update pipeline.py

pipeline.py

@@ -79,7 +79,7 @@ class PreTrainedPipeline():
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         # process input
-        #inputs = data.pop("inputs", data)
+        inputs = data.pop("inputs", data)
         # process input text
-        prediction = model_infer( self.model, self.tokenizer, data+"TL;DR")
+        prediction = model_infer( self.model, self.tokenizer, inputs['text']+"TL;DR")
         return {"text": prediction}