diff --git a/examples/streamlit_spacy.py b/examples/streamlit_spacy.py
index 1afa1cd32..a2da123c2 100644
--- a/examples/streamlit_spacy.py
+++ b/examples/streamlit_spacy.py
@@ -26,12 +26,12 @@ DEFAULT_TEXT = "Mark Zuckerberg is the CEO of Facebook."
HTML_WRAPPER = """
{}
"""
-@st.cache(ignore_hash=True)
+@st.cache(allow_output_mutation=True)
def load_model(name):
return spacy.load(name)
-@st.cache(ignore_hash=True)
+@st.cache(allow_output_mutation=True)
def process_text(model_name, text):
nlp = load_model(model_name)
return nlp(text)
@@ -79,7 +79,9 @@ if "ner" in nlp.pipe_names:
st.header("Named Entities")
st.sidebar.header("Named Entities")
label_set = nlp.get_pipe("ner").labels
- labels = st.sidebar.multiselect("Entity labels", label_set, label_set)
+ labels = st.sidebar.multiselect(
+ "Entity labels", options=label_set, default=list(label_set)
+ )
html = displacy.render(doc, style="ent", options={"ents": labels})
# Newlines seem to mess with the rendering
html = html.replace("\n", " ")