Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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='
|
| 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 |
|