paul hilders commited on
Commit
fcc28d6
·
1 Parent(s): 6d91375

Attempt to add image labels

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -90,10 +90,10 @@ def NER_demo(image, text):
90
  overlapped, highlighted_text = run_demo(image, text)
91
 
92
  # Then, we run the demo for each of the named entities:
93
- gallery_images = [overlapped]
94
  for ent_text, ent_label in highlighed_entities:
95
  overlapped_ent, highlighted_text_ent = run_demo(image, ent_text)
96
- gallery_images.append(overlapped_ent)
97
 
98
  return highlighed_entities, gallery_images
99
 
 
90
  overlapped, highlighted_text = run_demo(image, text)
91
 
92
  # Then, we run the demo for each of the named entities:
93
+ gallery_images = [(overlapped, "Default explanation")]
94
  for ent_text, ent_label in highlighed_entities:
95
  overlapped_ent, highlighted_text_ent = run_demo(image, ent_text)
96
+ gallery_images.append((overlapped_ent, ent_text))
97
 
98
  return highlighed_entities, gallery_images
99