aes

app.py

@@ -184,12 +184,13 @@ if __name__=='__main__':
         st.session_state['page_status'] = 'model_selection'
 
     if st.session_state['page_status']=='model_selection':
+        show_instruction('0. Select the model and click "Confirm"',fontsize=16)
         model_name = st.selectbox('Please select the model from below.',
                                     ('bert-base-uncased','bert-large-cased',
                                     'roberta-base','roberta-large',
-                                    'albert-base-v2','albert-large-v2','albert-xlarge-v2','albert-xxlarge-v2'),index=3)
-        st.session_state['model_name'] = model_name
-        if st.button('Confirm',key='models'):
+                                    'albert-base-v2','albert-large-v2','albert-xlarge-v2','albert-xxlarge-v2'),
+                                    index=3,key='model_name',label_visibility='collapsed')
+        if st.button('Confirm',key='confirm_models'):
             st.session_state['page_status'] = 'type_in'
             st.experimental_rerun()
 
@@ -213,11 +214,11 @@ if __name__=='__main__':
         sent_2 = st.session_state['sent_2']
 
         show_instruction('2. Select sites to mask out and click "Confirm"',fontsize=16)
-        show_instruction('------------------------------',fontsize=32)
+        #show_instruction('------------------------------',fontsize=32)
         annotate_mask(1,sent_1)
         show_instruction('------------------------------',fontsize=32)
         annotate_mask(2,sent_2)
-        if st.button('Confirm',key='mask'):
+        if st.button('Confirm',key='confirm_mask'):
             st.session_state['page_status'] = 'annotate_options'
             st.experimental_rerun()
 
@@ -226,11 +227,11 @@ if __name__=='__main__':
         sent_2 = st.session_state['sent_2']
 
         show_instruction('3. Select options and click "Confirm"',fontsize=16)
-        show_instruction('------------------------------',fontsize=32)
+        #show_instruction('------------------------------',fontsize=32)
         annotate_options(1,sent_1)
         show_instruction('------------------------------',fontsize=32)
         annotate_options(2,sent_2)
-        if st.button('Confirm',key='option'):
+        if st.button('Confirm',key='confirm_option'):
             st.session_state['page_status'] = 'analysis'
             st.experimental_rerun()