Update app.py

app.py

@@ -4,23 +4,23 @@ from transformers import pipeline
 unmasker = pipeline('fill-mask', model='dsfsi/zabantu-xlm-roberta')
 
 sample_sentences = {
-    'zulu': "Le ndoda ithi izo____ ukudla.",  
-    'tshivenda': "Mufana uyo____ vhukuma.",  
-    'sepedi': "Mosadi o ____ pheka.",  
-    'tswana': "Monna o ____ tsamaya.",  
-    'tsonga': "N'wana wa xisati u ____ ku tsaka."  
+    'zulu': "Le ndoda ithi izo <mask> ukudla.",  
+    'tshivenda': "Mufana uyo <mask> vhukuma.",  
+    'sepedi': "Mosadi o <mask> pheka.",  
+    'tswana': "Monna o <mask> tsamaya.",  
+    'tsonga': "N'wana wa xisati u <mask> ku tsaka."  
 }
 
 def fill_mask_for_languages(sentences):
     results = {}
     for language, sentence in sentences.items():
-        masked_sentence = sentence.replace('____', unmasker.tokenizer.mask_token)
+        masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
         unmasked = unmasker(masked_sentence)
         results[language] = unmasked 
     return results
 
 def replace_mask(sentence, predicted_word):
-    return sentence.replace("____", f"**{predicted_word}**")
+    return sentence.replace("<mask>", f"**{predicted_word}**")
 
 st.title("Fill Mask| Zabantu-XLM-Roberta")
 st.write(f"")
@@ -33,7 +33,7 @@ with col1:
     ))
 
     if st.button("Submit"):
-        user_masked_sentence = user_sentence.replace('____', unmasker.tokenizer.mask_token)
+        user_masked_sentence = user_sentence.replace('<mask>', unmasker.tokenizer.mask_token)
 
 with col2:
     if 'user_masked_sentence' in locals():