diff --git a/components/GetSpacyData.py b/components/GetSpacyData.py index 6fcfef8..835c98b 100644 --- a/components/GetSpacyData.py +++ b/components/GetSpacyData.py @@ -108,8 +108,8 @@ def GetEntities(doc) -> List[Entity]: entities.append( Entity( name=entity.text, - startIndex=entity.start_char, - endIndex=entity.end_char, + startIndex=entity.start_char-entity.sent.start_char, + endIndex=entity.end_char-entity.sent.start_char, sentence=entity.sent.text, sentenceStartIndex=entity.sent.start_char, sentenceEndIndex=entity.sent.end_char,