diff --git a/REL/mention_detection.py b/REL/mention_detection.py index 4ffa95e..89c184b 100644 --- a/REL/mention_detection.py +++ b/REL/mention_detection.py @@ -146,8 +146,8 @@ def find_mentions(self, dataset, tagger=None): ): text, start_pos, end_pos, conf, tag = ( entity.text, - entity.start_pos, - entity.end_pos, + entity.start_position, + entity.end_position, entity.score, entity.tag, ) diff --git a/requirements.txt b/requirements.txt index 6fa0ed4..ac60754 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,8 @@ colorama konoha -flair +flair>=0.11 segtok Pillow>=7.1.0 torch nltk -anyascii \ No newline at end of file +anyascii