Update app.py

app.py

@@ -23,7 +23,7 @@ col1, col2 = st.columns(2)
 
 with col1:
     with st.container(border=True):
-        sample_sentence = "Vhana vhane vha kha ḓi bva u bebwa vha kha khombo ya u <mask> nga Listeriosis"
+        sample_sentence = "Vhana vhane vha kha ḓi bva u bebwa vha kha khombo ya u <mask> nga Listeriosis."
     
         if "text_input" not in st.session_state:
             st.session_state.text_area = ""
@@ -38,26 +38,28 @@ with col1:
         if st.button("Submit"):
             result = fill_mask(input_sentences)
     
-        st.write("Examples")
+        st.markdown("Examples")
         st.code(sample_sentence, wrap_lines=True)
 
 with col2:
-    if 'result' in locals():  
-        if result:
-            for sentence, predictions in result.items():
-                for prediction in predictions:
-                    predicted_word = prediction['token_str']
-                    score = prediction['score'] * 100
-
-                    st.markdown(f"""
-                    <div class="bar">
-                        <div class="bar-fill" style="width: {score}%;"></div>
-                    </div>
-                    <div class="container">
-                        <div style="align-items: left;">{predicted_word}</div>
-                        <div style="align-items: right;">{score:.2f}%</div>
-                    </div>
-                    """, unsafe_allow_html=True)
+    with st.container(border=True):
+        st.markdown("Output")
+        if 'result' in locals():  
+            if result:
+                for sentence, predictions in result.items():
+                    for prediction in predictions:
+                        predicted_word = prediction['token_str']
+                        score = prediction['score'] * 100
+    
+                        st.markdown(f"""
+                        <div class="bar">
+                            <div class="bar-fill" style="width: {score}%;"></div>
+                        </div>
+                        <div class="container">
+                            <div style="align-items: left;">{predicted_word}</div>
+                            <div style="align-items: right;">{score:.2f}%</div>
+                        </div>
+                        """, unsafe_allow_html=True)
 
 if 'result' in locals():  
         if result: