debug
Browse files- 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]) ->
|
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`)
|