relik-ie commited on
Commit
7d340b1
1 Parent(s): 8432c73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -236,7 +236,7 @@ def get_span_annotations(response, doc, ner=False):
236
  span.end,
237
  el_link_wrapper.format(
238
  span.label.replace(" ", "_"), span.label
239
- ) if span.label != "--NME--" and not ner else span.label,
240
  # kb_id=span.label.replace(" ", "_")
241
  )
242
  )
@@ -269,7 +269,7 @@ def generate_graph(spans, response, colors, dict_ents, bgcolor="#111827", font_c
269
 
270
  for ent in spans:
271
  # if not NME use title:
272
- if dict_ents[(ent.start, ent.end)][2] != "--NME--" or not ner:
273
  g.add_node(
274
  dict_ents[(ent.start, ent.end)][2],
275
  label=dict_ents[(ent.start, ent.end)][2],
 
236
  span.end,
237
  el_link_wrapper.format(
238
  span.label.replace(" ", "_"), span.label
239
+ ) if (span.label != "--NME--" and not ner) else span.label,
240
  # kb_id=span.label.replace(" ", "_")
241
  )
242
  )
 
269
 
270
  for ent in spans:
271
  # if not NME use title:
272
+ if dict_ents[(ent.start, ent.end)][2] != "--NME--" and not ner:
273
  g.add_node(
274
  dict_ents[(ent.start, ent.end)][2],
275
  label=dict_ents[(ent.start, ent.end)][2],