Alexandre-Numind commited on
Commit
3d252ed
·
1 Parent(s): 7011838

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
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['entity'] == current['entity'] and (next_entity['start'] == current['end'] + 1 or next_entity['start'] == current['end']):
49
- current['word'] += ' ' + next_entity['word']
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(