impresso-project
/

ner-stacked-bert-multilingual

Token Classification

Model card Files Files and versions

emanuelaboros commited on Oct 23, 2024

Commit

d214adc

·

1 Parent(s): 0968912

add comp back

Files changed (1) hide show

generic_ner.py +2 -1

generic_ner.py CHANGED Viewed

@@ -524,7 +524,8 @@ def remove_trailing_stopwords(entities):
                 entities.remove(entity)
             else:
                 new_entities.append(entity)
     if DEBUG:
         print(f"Remained entities: {len(new_entities)}")
     return new_entities

                 entities.remove(entity)
             else:
                 new_entities.append(entity)
+        else:
+            new_entities.append(entity)
     if DEBUG:
         print(f"Remained entities: {len(new_entities)}")
     return new_entities