Update pipeline.py

pipeline.py

@@ -76,10 +76,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[Any, Any]) -> Dict[Any, Any]:
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         # process input
         inputs = data.pop("inputs", data)
         # process input text
         prediction = model_infer(self.model, self.tokenizer, inputs+"TL;DR")
-        return {"text": prediction}
+        return [{"text": prediction}]