nickmuchi commited on
Commit
040ddf0
1 Parent(s): 08fa659

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -167,7 +167,7 @@ def get_all_entities_per_sentence(text):
167
  # entities_this_sentence.append(entity.text)
168
 
169
  # XLM ENTITIES
170
- entities_xlm = [entity["word"] for entity in ner_model(str(sentence))]
171
  for entity in entities_xlm:
172
  entities_this_sentence.append(str(entity))
173
 
 
167
  # entities_this_sentence.append(entity.text)
168
 
169
  # XLM ENTITIES
170
+ entities_xlm = [entity["word"] for entity in ner_pipe(str(sentence))]
171
  for entity in entities_xlm:
172
  entities_this_sentence.append(str(entity))
173