Wootang01 commited on
Commit
e4016f5
1 Parent(s): 41c9a59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ st.set_page_config(layout="wide", page_title="Vocabulary Categorizer")
17
  st.title("Vocabulary Categorizer")
18
  st.write("This application identifies, highlights and categories nouns.")
19
 
20
- model_list = ['xlm-roberta-large-finetuned-conll03-english', 'xlm-roberta-large-xnli']
21
 
22
  st.sidebar.header("Vocabulary categorizer")
23
  model_checkpoint = st.sidebar.radio("", model_list)
@@ -32,7 +32,7 @@ if model_checkpoint == "xlm-roberta-large-finetuned-conll03-english":
32
  aggregation = st.sidebar.radio("", ('simple', 'none'))
33
  st.sidebar.write(xlm_agg_strategy_info)
34
  st.sidebar.write("")
35
- elif model_checkpoint == "xlm-roberta-large-xnli":
36
  aggregation = st.sidebar.radio("", ('simple', 'none'))
37
  st.sidebar.write(xlm_agg_strategy_info)
38
  st.sidebar.write("")
 
17
  st.title("Vocabulary Categorizer")
18
  st.write("This application identifies, highlights and categories nouns.")
19
 
20
+ model_list = ['xlm-roberta-large-finetuned-conll03-english', 'xlm-roberta-large']
21
 
22
  st.sidebar.header("Vocabulary categorizer")
23
  model_checkpoint = st.sidebar.radio("", model_list)
 
32
  aggregation = st.sidebar.radio("", ('simple', 'none'))
33
  st.sidebar.write(xlm_agg_strategy_info)
34
  st.sidebar.write("")
35
+ elif model_checkpoint == "xlm-roberta-large":
36
  aggregation = st.sidebar.radio("", ('simple', 'none'))
37
  st.sidebar.write(xlm_agg_strategy_info)
38
  st.sidebar.write("")