Update pipeline.py

pipeline.py

@@ -77,7 +77,7 @@ class PreTrainedPipeline():
         self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
 
 
-    def __call__(self, Dict[str, str]) -> Dict[str, str]:
+    def __call__(self, Dict[Any, Any]) -> Dict[Any, Any]:
         # process input
         inputs = data.pop("inputs", data)
         # process input text