Shad0ws commited on
Commit
65beb69
·
1 Parent(s): f8f5be1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -54,3 +54,12 @@ if input_text or button_clicked:
54
  st.write("Named Entities:")
55
  st.write(", ".join([f"{ent.text}:{ent.label_}" for ent in entities.ents if "#" not in ent.text and ent.label_ in ["PERSON", "ORG"]]))
56
  st.write("")
 
 
 
 
 
 
 
 
 
 
54
  st.write("Named Entities:")
55
  st.write(", ".join([f"{ent.text}:{ent.label_}" for ent in entities.ents if "#" not in ent.text and ent.label_ in ["PERSON", "ORG"]]))
56
  st.write("")
57
+
58
+ with st.expander("Named Entities"):
59
+ ent_list = []
60
+ for entity in entities.ents:
61
+ if "#" not in entity.text and entity.label_ in ["PERSON", "ORG"]:
62
+ ent_list.append(entity.text+" : "+entity.label_)
63
+ ent_set = set(ent_list)
64
+ ent_string = ', '.join(list(ent_set))
65
+ st.write(ent_string)