adorkin commited on
Commit
57c5f38
·
1 Parent(s): 054f7dc

Update models

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -48,7 +48,10 @@ def main():
48
  message = st.text_area('Or type a sentence to see if our AL Algorithm can detect your emotion', example)
49
  # st.title(message)
50
  st.text('')
51
- models_to_choose = ["AlekseyDorkin/xlm-roberta-en-ru-emoji"]
 
 
 
52
 
53
  BASE_MODEL = st.selectbox("Choose a model", models_to_choose)
54
 
 
48
  message = st.text_area('Or type a sentence to see if our AL Algorithm can detect your emotion', example)
49
  # st.title(message)
50
  st.text('')
51
+ models_to_choose = [
52
+ "amazon-sagemaker-community/xlm-roberta-en-ru-emoji-v2",
53
+ "AlekseyDorkin/xlm-roberta-en-ru-emoji"
54
+ ]
55
 
56
  BASE_MODEL = st.selectbox("Choose a model", models_to_choose)
57