Update pipeline.py

pipeline.py

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