taka-yamakoshi commited on
Commit
9839e32
1 Parent(s): f5b635d
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -189,7 +189,8 @@ if __name__=='__main__':
189
  ('bert-base-uncased','bert-large-cased',
190
  'roberta-base','roberta-large',
191
  'albert-base-v2','albert-large-v2','albert-xlarge-v2','albert-xxlarge-v2'),
192
- index=3,key='model_name',label_visibility='collapsed')
 
193
  if st.button('Confirm',key='confirm_models'):
194
  st.session_state['page_status'] = 'type_in'
195
  st.experimental_rerun()
 
189
  ('bert-base-uncased','bert-large-cased',
190
  'roberta-base','roberta-large',
191
  'albert-base-v2','albert-large-v2','albert-xlarge-v2','albert-xxlarge-v2'),
192
+ index=3,label_visibility='collapsed')
193
+ st.session_state['model_name'] = model_name
194
  if st.button('Confirm',key='confirm_models'):
195
  st.session_state['page_status'] = 'type_in'
196
  st.experimental_rerun()