Spaces:
Running
on
T4
Running
on
T4
Commit
·
3d252ed
1
Parent(s):
7011838
Update app.py
Browse files
app.py
CHANGED
@@ -40,21 +40,16 @@ examples = [
|
|
40 |
def merge_entities(entities):
|
41 |
if not entities:
|
42 |
return []
|
43 |
-
|
44 |
merged = []
|
45 |
current = entities[0]
|
46 |
-
|
47 |
for next_entity in entities[1:]:
|
48 |
-
if next_entity['
|
49 |
-
current['
|
50 |
current['end'] = next_entity['end']
|
51 |
else:
|
52 |
merged.append(current)
|
53 |
current = next_entity
|
54 |
-
|
55 |
-
# Append the last entity
|
56 |
merged.append(current)
|
57 |
-
|
58 |
return merged
|
59 |
|
60 |
def ner(
|
|
|
40 |
def merge_entities(entities):
|
41 |
if not entities:
|
42 |
return []
|
|
|
43 |
merged = []
|
44 |
current = entities[0]
|
|
|
45 |
for next_entity in entities[1:]:
|
46 |
+
if next_entity['label'] == current['label'] and (next_entity['start'] == current['end'] + 1 or next_entity['start'] == current['end']):
|
47 |
+
current['text'] = text[current['start']: next_entity['end']].strip()
|
48 |
current['end'] = next_entity['end']
|
49 |
else:
|
50 |
merged.append(current)
|
51 |
current = next_entity
|
|
|
|
|
52 |
merged.append(current)
|
|
|
53 |
return merged
|
54 |
|
55 |
def ner(
|