Spaces:

amsterdamNLP
/

CLIP-attention-rollout

Runtime error

paul hilders commited on Jul 3, 2022

Commit

82a35b6

•

1 Parent(s): 929c841

Test highlighting

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ def NER_demo(image, text):
     for ent in nlp(text).ents:
         ent_text = ent.text
         ent_label = ent.label_
-        highlighed_entities.append((ent_text, ent_label))
     print(highlighed_entities)

     for ent in nlp(text).ents:
         ent_text = ent.text
         ent_label = ent.label_
+        highlighed_entities.append((ent_text, 1.0))
     print(highlighed_entities)