Update app.py

app.py

@@ -17,7 +17,7 @@ st.set_page_config(layout="wide", page_title="Vocabulary Categorizer")
 st.title("Vocabulary Categorizer")
 st.write("This application identifies, highlights and categories nouns.")
 
-model_list = ['xlm-roberta-large-finetuned-conll03-english', 'spacy/en_core_web_sm']
+model_list = ['xlm-roberta-large-finetuned-conll03-english', 'xlm-roberta-base']
 
 st.sidebar.header("Vocabulary categorizer")
 model_checkpoint = st.sidebar.radio("", model_list)
@@ -32,8 +32,10 @@ if model_checkpoint == "xlm-roberta-large-finetuned-conll03-english":
     aggregation = st.sidebar.radio("", ('simple', 'none'))
     st.sidebar.write(xlm_agg_strategy_info)
     st.sidebar.write("")
-elif model_checkpoint == "en_core_web_sm": 
-    aggregation = 'none'
+elif model_checkpoint == "xlm-roberta-base": 
+    aggregation = st.sidebar.radio("", ('simple', 'none'))
+    st.sidebar.write(xlm_agg_strategy_info)
+    st.sidebar.write("")
 
 st.subheader("Select Text Input Method")
 input_method = st.radio("", ('Select from Examples', 'Write or Paste New Text'))