Update pipeline.py

pipeline.py

@@ -77,9 +77,9 @@ class PreTrainedPipeline():
         self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
 
 
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+    def __call__(self, data) -> 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, inputs['text']+"TL;DR")
+        prediction = model_infer( self.model, self.tokenizer, data+"TL;DR")
         return {"text": prediction}