Update

app.py

@@ -23,7 +23,7 @@ st.sidebar.write("")
 
 
 st.sidebar.header("Select type of PERSON detection")
-named_persons_only = st.sidebar.radio("", ('Proper nouns only', 'All nouns'))
+named_persons_only_radio = st.sidebar.radio("", ('Proper nouns only', 'All nouns'))
 
 st.sidebar.write("Types of entities detected: 'PERSON', 'ORG', 'GPE', 'LOC', 'NAT_REL_POL', 'EVENT', 'LANGUAGE', 'WORK_OF_ART', 'DATETIME', 'PERIOD', 'MONEY', 'QUANTITY', 'NUMERIC', 'ORDINAL', 'FACILITY'")
 
@@ -39,7 +39,7 @@ elif input_method == "Write or Paste New Text":
 
 @st.cache(allow_output_mutation=True)
 def setModel(named_persons_only):
-    ner = roner.NER(named_persons_only=True if named_persons_only.startswith("Proper") else False)
+    ner = roner.NER(named_persons_only=named_persons_only)
     return ner
 
 @st.cache(allow_output_mutation=True)
@@ -51,7 +51,7 @@ def get_html(html: str):
 Run_Button = st.button("Run", key=None)
 if Run_Button == True:
     
-    ner = setModel(named_persons_only = False)
+    ner = setModel(named_persons_only = True if named_persons_only_radio.startswith("Proper") else False)
     output = ner(input_text)[0] # only one sentence
 
     # tabular form