chicham commited on
Commit
70861a8
·
unverified ·
1 Parent(s): 54ab090

Modify aggregation strategy for NER (#2)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -253,7 +253,7 @@ def main():
253
  load_fn=lambda: pipeline(
254
  "ner",
255
  model=cfg.ner.general,
256
- aggregation_strategy=AggregationStrategy.SIMPLE,
257
  ),
258
  ),
259
  recipe=Predictor(
 
253
  load_fn=lambda: pipeline(
254
  "ner",
255
  model=cfg.ner.general,
256
+ aggregation_strategy=AggregationStrategy.MAX,
257
  ),
258
  ),
259
  recipe=Predictor(