fix return type of the [_extract_ner] function
Browse files
README.md
CHANGED
@@ -78,7 +78,7 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification, pipelin
|
|
78 |
transformers_logger = logging.getLogger("transformers")
|
79 |
transformers_logger.setLevel(logging.WARNING)
|
80 |
|
81 |
-
def _extract_ner(sent: str, ner: pipeline) ->
|
82 |
grouped_ents = []
|
83 |
current_ent = {}
|
84 |
|
|
|
78 |
transformers_logger = logging.getLogger("transformers")
|
79 |
transformers_logger.setLevel(logging.WARNING)
|
80 |
|
81 |
+
def _extract_ner(sent: str, ner: pipeline) -> list:
|
82 |
grouped_ents = []
|
83 |
current_ent = {}
|
84 |
|