Update app.py

app.py

@@ -54,12 +54,13 @@ with col1:
         button1, button2, _ = st.columns([2, 2, 4])
         with button1:
             if st.button("Test Example"):
-                result, warnings = fill_mask(sample_sentence)  # Passing the correct dictionary format
+                input_sentences = sample_sentence
+                result, warnings = fill_mask(input_sentences) 
                 st.session_state['text_input'] = "\n".join([f"{lang}: {sentence}" for lang, sentence in sample_sentence.items()])
         
         with button2:
             if st.button("Submit"):
-                result, warnings = fill_mask(input_sentences)  # input_sentences is already a dictionary
+                result, warnings = fill_mask(input_sentences) 
                 st.session_state['warnings'] = warnings  
     
         if st.session_state['warnings']:
@@ -87,11 +88,19 @@ with col2:
                         <div class="bar-fill" style="width: {score}%;"></div>
                     </div>
                     <div class="container">
-                        <div style="align-items: left;">{predicted_word} ({language})</div>
-                        <div style="align-items: right;">{score:.2f}%</div>
+                        <div style="align-items: left;">{predicted_word}({language})</div>
+                        <div style="align-items: center;">{score:.2f}%</div>
                     </div>
                     """, unsafe_allow_html=True)
 
+if 'predictions' in locals():  
+    if predictions:
+        for language, language_predictions in predictions.items():
+            original_sentence = sample_sentences[language]
+            predicted_sentence = replace_mask(original_sentence, language_predictions[0]['token_str'])  
+            # st.write(language_predictions)
+            # st.write(f"Original sentence ({language}): {original_sentence}")
+            st.write(f"{language}: {predicted_sentence}\n")
 css = """
 <style>
 footer {display:none !important;}