Update app.py

app.py

@@ -34,7 +34,7 @@ if 'warnings' not in st.session_state:
     st.session_state['warnings'] = []
 
 with col1:
-    with st.container():  # Removed the border=True argument which is invalid for st.container
+    with st.container(border=True): 
         st.markdown("Input :clipboard:")
         sample_sentence = {
             'zulu': "Le ndoda ithi izo <mask> ukudla.",  
@@ -54,13 +54,13 @@ with col1:
         button1, button2, _ = st.columns([2, 2, 4])
         with button1:
             if st.button("Test Example"):
+                result, warnings = fill_mask("\n".join([f"{lang}: {sentence}" for lang, sentence in sample_sentence.items()]))
                 st.session_state['text_input'] = "\n".join([f"{lang}: {sentence}" for lang, sentence in sample_sentence.items()])
-                st.experimental_rerun()  # Rerun to update the text_area with example sentences
         
         with button2:
             if st.button("Submit"):
                 result, warnings = fill_mask(input_sentences)
-                st.session_state['warnings'] = warnings  # Update warnings in session state
+                st.session_state['warnings'] = warnings  
     
         if st.session_state['warnings']:
             for warning in st.session_state['warnings']:
@@ -70,7 +70,7 @@ with col1:
         st.code(sample_sentence, wrap_lines=True)
 
 with col2:
-    with st.container():  # Removed the border=True argument which is invalid for st.container
+    with st.container(border=True): 
         st.markdown("Output :bar_chart:")
         if input_sentences:
             for language, sentence in input_sentences.items():