Update app.py

app.py

@@ -15,16 +15,17 @@ def fill_mask(sentences):
             warnings.append(f"Warning: Choose language for {sentence}")
             continue
             
-        if language != 'choose language' and sentence == '':
+        if language != 'choose language' and sentence == "":
             warnings.append(f"Warning: Enter sentence for {language}")
             continue
         
         if "<mask>" in sentence:
             masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
             unmasked = unmasker(masked_sentence)
-            results[language] = unmasked
+            results[key] = (unmasked,language,sentence)
         else:
             warnings.append(f"Warning: No <mask> token found in sentence: {sentence}")
+    
     return results, warnings
 
 def replace_mask(sentence, predicted_word):
@@ -122,7 +123,7 @@ with col2:
         st.markdown("Output :bar_chart:")
         if 'result' in locals() and result:  
             if len(result) == 1:
-                for language, predictions in result.items():
+                for key,(predictions, language, sentence) in result.items():
                     for prediction in predictions:
                         predicted_word = prediction['token_str']
                         score = prediction['score'] * 100
@@ -158,7 +159,7 @@ with col2:
 if 'result' in locals():  
     if result:
         line = 0
-        for sentence, predictions in result.items():
+        for key,(predictions, language, sentence) in result.items():
             line += 1
             predicted_word = predictions[0]['token_str']
             full_sentence = replace_mask(sentence, predicted_word)