input type

pipeline.py

@@ -17,7 +17,7 @@ class PreTrainedPipeline:
         self.predictor = CorefPredictor.from_extracted_archive(archive_content)
         self.predictor.set_device("cpu")
 
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+    def __call__(self, data: str) -> Dict[str, Any]:
         """
          data args:
               inputs (:obj: `str`)
@@ -26,7 +26,5 @@ class PreTrainedPipeline:
               A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
-        return data
-        inputs: str = data.pop("inputs", data)
-        prediction = self.predictor.resolve_paragraphs(inputs.split("\n\n"))
+        prediction = self.predictor.resolve_paragraphs(data.split("\n\n"))
         return prediction.to_dict()