Make the default NER labels less model-specific [ci skip] (#4361)

This commit is contained in:
adrianeboyd 2019-10-02 12:05:17 +02:00 committed by Ines Montani
parent dda86118bd
commit d82241218a

View File

@ -78,9 +78,8 @@ if "parser" in nlp.pipe_names:
if "ner" in nlp.pipe_names:
st.header("Named Entities")
st.sidebar.header("Named Entities")
default_labels = ["PERSON", "ORG", "GPE", "LOC"]
labels = st.sidebar.multiselect(
"Entity labels", nlp.get_pipe("ner").labels, default_labels
"Entity labels", nlp.get_pipe("ner").labels, nlp.get_pipe("ner").labels
)
html = displacy.render(doc, style="ent", options={"ents": labels})
# Newlines seem to mess with the rendering