Update app.py

app.py

@@ -12,7 +12,7 @@ def fill_mask(sentences):
         if "<mask>" in sentence:
             masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
             unmasked = unmasker(masked_sentence)
-            results[language] = unmasked 
+            results[language] = unmasked
         else:
             warnings.append(f"Warning: No <mask> token found in sentence: {sentence}")
     return results, warnings
@@ -60,7 +60,7 @@ with col1:
         
         with button2:
             if st.button("Submit"):
-                result, warnings = fill_mask(input_sentences) 
+                result, warnings = fill_mask(input_sentences)
                 st.session_state['warnings'] = warnings  
     
         if st.session_state['warnings']:
@@ -88,19 +88,18 @@ 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: center;">{score:.2f}%</div>
+                        <div style="align-items: left;">{predicted_word} ({language})</div>
+                        <div style="align-items: right;">{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]
+    if result:
+        for language, language_predictions in result.items():  # result is a dictionary
+            original_sentence = sample_sentence[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;}
@@ -113,7 +112,6 @@ footer {display:none !important;}
     width: 100%;
 }
 .bar {
-    # width: 70%;
     background-color: #e6e6e6;
     border-radius: 12px;
     overflow: hidden;