Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import pipeline
 
-unmasker = pipeline('fill-mask', model='dsfsi/zabantu-bantu-250m')
+unmasker = pipeline('fill-mask', model='dsfsi/zabantu-xlm-roberta')
 
 sample_sentences = {
     'Zulu': "Le ndoda ithi izo____ ukudla.",  
@@ -24,11 +24,12 @@ def fill_mask_for_languages(sentences):
 def replace_mask(sentence, predicted_word):
     return sentence.replace("____", predicted_word)
 
-st.title("Fill Mask for Multiple Languages | Zabantu-Bantu-250m")
+st.title("Fill Mask for Multiple Languages | Zabantu-XLM-Roberta")
 st.write("This app predicts the missing word for sentences in Zulu, Tshivenda, Sepedi, Tswana, and Tsonga using a Zabantu BERT model.")
 
 user_sentence = st.text_input("Enter your own sentence with a masked word (use '____'):", "\n".join(
-        f"'{lang}': '{sentence}',\n"
+        f"'{lang}': '{sentence}',"
+        f"\n"
         for lang, sentence in sample_sentences.items()
     ))