Update app.py

app.py

@@ -21,9 +21,9 @@ def replace_mask(sentence, predicted_word):
     return sentence.replace("<mask>", f"**{predicted_word}**")
 
 st.title("Fill Mask | Zabantu-XLM-Roberta")
-st.write(f"")
+st.write("")
 
-st.markdown("Zabantu-XLMR refers to a fleet of models trained on different combinations of South African Bantu languages. These include: Zabantu-VEN, Zabantu-NSO, Zabantu-NSO+VEN, Zabantu-SOT+VEN, Zabantu-BANTU(from 9 South African Bantu languages)")
+st.markdown("Zabantu-XLMR refers to a fleet of models trained on different combinations of South African Bantu languages.")
 
 col1, col2 = st.columns(2)
 
@@ -49,17 +49,17 @@ with col1:
             value=st.session_state['text_input']
         )
     
-        input_sentences = {f"sentence_{i}": sentence for i, sentence in enumerate(text_input.split("\n")) if sentence} 
+        input_sentences = {f"sentence_{i}": sentence for i, sentence in enumerate(text_input.split("\n")) if sentence}
 
         button1, button2, _ = st.columns([2, 2, 4])
         with button1:
             if st.button("Test Example"):
-                result, warnings = fill_mask("\n".join([f"{lang}: {sentence}" for lang, sentence in sample_sentence.items()]))
+                result, warnings = fill_mask(sample_sentence)  # Passing the correct dictionary format
                 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)
+                result, warnings = fill_mask(input_sentences)  # input_sentences is already a dictionary
                 st.session_state['warnings'] = warnings  
     
         if st.session_state['warnings']: