kwang2049 commited on
Commit
5e0e65c
1 Parent(s): 3d594a8
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -17,7 +17,7 @@ class PreTrainedPipeline:
17
  self.predictor = CorefPredictor.from_extracted_archive(archive_content)
18
  self.predictor.set_device("cpu")
19
 
20
- def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
21
  """
22
  data args:
23
  inputs (:obj: `str`)
 
17
  self.predictor = CorefPredictor.from_extracted_archive(archive_content)
18
  self.predictor.set_device("cpu")
19
 
20
+ def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
21
  """
22
  data args:
23
  inputs (:obj: `str`)