mrfirdauss commited on
Commit
fb81479
·
verified ·
1 Parent(s): bfb991c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def process_tokens(tokens, tag_prefix):
113
  current_entity = {}
114
  current_entity['text'] = token
115
  current_entity['type'] = tag
116
- elif tag.startswith('I-') and tag.endswith('GPA') and current_entity:
117
  current_entity['text'] += '' + token
118
  elif tag.startswith('I-') and tag.endswith(tag_prefix) and current_entity:
119
  # Continue the current entity
 
113
  current_entity = {}
114
  current_entity['text'] = token
115
  current_entity['type'] = tag
116
+ elif tag.startswith('I-') and (tag.endswith('GPA') or tag.endswith('URL')) and current_entity:
117
  current_entity['text'] += '' + token
118
  elif tag.startswith('I-') and tag.endswith(tag_prefix) and current_entity:
119
  # Continue the current entity