asahi417 commited on
Commit
6daa310
·
1 Parent(s): 0cf1f01
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def predict(text):
34
  "text": text,
35
  "ents": [{
36
  "start": retain_char_position(entity['position'][0]),
37
- "end": retain_char_position(entity['position'][-1]) + 1 + len(entity['entity'][-1]),
38
  "label": entity['type']
39
  } for entity in output['entity_prediction'][0]],
40
  "title": None
 
34
  "text": text,
35
  "ents": [{
36
  "start": retain_char_position(entity['position'][0]),
37
+ "end": retain_char_position(entity['position'][-1]) + len(entity['entity'][-1]),
38
  "label": entity['type']
39
  } for entity in output['entity_prediction'][0]],
40
  "title": None