tweak

app.py

@@ -65,12 +65,13 @@ if __name__=='__main__':
         mask_id = tokenizer('[MASK]').input_ids[1:-1][0]
 
         st.write('1. Type in the sentences and click "Tokenize"')
-        sent_1 = st.sidebar.text_input('Sentence 1',value='It is better to play a prank on Samuel than Craig because he gets angry less often.')
-        sent_2 = st.sidebar.text_input('Sentence 2',value='It is better to play a prank on Samuel than Craig because he gets angry more often.')
-        if st.sidebar.button('Tokenize'):
+        sent_1 = st.text_input('Sentence 1',value='It is better to play a prank on Samuel than Craig because he gets angry less often.')
+        sent_2 = st.text_input('Sentence 2',value='It is better to play a prank on Samuel than Craig because he gets angry more often.')
+        if st.button('Tokenize'):
             st.session_state['page_status'] = 'tokenized'
             st.session_state['sent_1'] = sent_1
             st.session_state['sent_2'] = sent_2
+            st.experimental_rerun()
 
     if st.session_state['page_status']=='tokenized':
         tokenizer,model = load_model()