Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/streamlit_spacy.py
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