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(
|