Update app.py

app.py

@@ -43,8 +43,8 @@ def merge_entities(entities):
     merged = []
     current = entities[0]
     for next_entity in entities[1:]:
-        if next_entity['label'] == current['label'] and (next_entity['start'] == current['end'] + 1 or next_entity['start'] == current['end']):
-            current['text'] = text[current['start']: next_entity['end']].strip()
+        if next_entity['entity'] == current['entity'] and (next_entity['start'] == current['end'] + 1 or next_entity['start'] == current['end']):
+            current['word'] = text[current['start']: next_entity['end']].strip()
             current['end'] = next_entity['end']
         else:
             merged.append(current)