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