DeDeckerThomas commited on
Commit
d9a0bbc
1 Parent(s): 23e4720

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -88,9 +88,9 @@ class KeyphraseExtractionPipeline(TokenClassificationPipeline):
88
  **kwargs
89
  )
90
 
91
- def postprocess(self, model_outputs):
92
  results = super().postprocess(
93
- model_outputs=model_outputs,
94
  aggregation_strategy=AggregationStrategy.SIMPLE,
95
  )
96
  return np.unique([result.get("word").strip() for result in results])
88
  **kwargs
89
  )
90
 
91
+ def postprocess(self, all_outputs):
92
  results = super().postprocess(
93
+ all_outputs=all_outputs,
94
  aggregation_strategy=AggregationStrategy.SIMPLE,
95
  )
96
  return np.unique([result.get("word").strip() for result in results])