Update app.py

app.py

@@ -9,9 +9,10 @@ st.set_page_config(layout="wide")
 def fill_mask(sentences):
     results = {}
     warnings = []
-    for sentence in sentences:
+    for (language, sentence) in sentences.items():
         if "<mask>" in sentence:
-            unmasked = unmasker(sentence)
+            masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
+            unmasked = unmasker(masked_sentence)
             results[sentence] = (language, unmasked)
         else:
             warnings.append(f"Warning: No <mask> token found in sentence: {sentence}")