davda54 commited on
Commit
ee95081
Β·
1 Parent(s): aeabbe6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -13,6 +13,7 @@ def parse(text):
13
 
14
  dependency_tree = render_dependency_tree(output["forms"], output["heads"], output["deprel"])
15
  table = render_table(output["forms"], output["lemmas"], output["upos"], output["xpos"], output["feats"], output["ne"])
 
16
 
17
  return dependency_tree, table
18
 
@@ -139,12 +140,14 @@ def render_table(forms, lemmas, upos, xpos, feats, named_entities):
139
  for i, ne in enumerate(named_entities):
140
  if ne == "O":
141
  named_entities_converted.append("")
 
 
142
  elif ne.startswith("B"):
143
- named_entities_converted.append(f"<< {ne.split('-')[1]}")
144
  elif ne.startswith("I") and i + 1 < len(named_entities) and named_entities[i + 1].startswith("I"):
145
- named_entities_converted.append(ne.split('-')[1])
146
  else:
147
- named_entities_converted.append(f"{ne.split('-')[1]} >>")
148
 
149
  array = [
150
  [""] + forms,
 
13
 
14
  dependency_tree = render_dependency_tree(output["forms"], output["heads"], output["deprel"])
15
  table = render_table(output["forms"], output["lemmas"], output["upos"], output["xpos"], output["feats"], output["ne"])
16
+ table = gr.DataFrame(**table, interactive=False, datatype="markdown")
17
 
18
  return dependency_tree, table
19
 
 
140
  for i, ne in enumerate(named_entities):
141
  if ne == "O":
142
  named_entities_converted.append("")
143
+ elif ne.startswith("B") and (i + 1 == len(named_entities) or named_entities[i + 1].startswith("I")):
144
+ named_entities_converted.append(f"<<β€” {ne.split('-')[1]} β€”")
145
  elif ne.startswith("B"):
146
+ named_entities_converted.append(f"<<β€” {ne.split('-')[1]} β€”>>")
147
  elif ne.startswith("I") and i + 1 < len(named_entities) and named_entities[i + 1].startswith("I"):
148
+ named_entities_converted.append("β€”β€”β€”β€”")
149
  else:
150
+ named_entities_converted.append(f"β€”β€”>>")
151
 
152
  array = [
153
  [""] + forms,