From ff8e71f46d8ee52a0ec94d973c2bcb87f57c563d Mon Sep 17 00:00:00 2001
From: Christos Aridas <chkoar@users.noreply.github.com>
Date: Sat, 15 Feb 2020 16:49:09 +0200
Subject: [PATCH] Update streamlit app (#5017)

* Update streamlit app [ci skip]

* Add all labels by default

* Tidy up and auto-format

Co-authored-by: Ines Montani <ines@ines.io>
---
 examples/streamlit_spacy.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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 = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
 
 
-@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", " ")