bakrianoo commited on
Commit
1964770
1 Parent(s): 90a87ad

fix return type of the [_extract_ner] function

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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) -> str:
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