Mardiyyah commited on
Commit
69e5789
·
verified ·
1 Parent(s): aed7757

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def load_model(model_name):
77
  return flair_pipes[model_name], "flair"
78
  else:
79
  if not model_name in hf_pipes:
80
- hf_pipes[model_name] = pipeline("ner", model=MODEL_REGISTRY[model_name], aggregation_strategy='simple')
81
  return hf_pipes[model_name], "hf"
82
 
83
 
 
77
  return flair_pipes[model_name], "flair"
78
  else:
79
  if not model_name in hf_pipes:
80
+ hf_pipes[model_name] = pipeline("ner", model=MODEL_REGISTRY[model_name], aggregation_strategy='first')
81
  return hf_pipes[model_name], "hf"
82
 
83