Update app.py

app.py

@@ -19,9 +19,11 @@ def fill_mask(sentences):
 def replace_mask(sentence, predicted_word):
  return sentence.replace("<mask>", f"**{predicted_word}**")
 
-st.title("Fill Mask | Zabantu-sot-ven-170m")
+st.title("Fill Mask | Zabantu-nso-ven-170m")
 st.write(f"")
 
+st.markdown("This is a variant of Zabantu pre-trained on a multilingual dataset of Tshivenda(ven) and Sepedi(nso) sentences on a transformer network with 170 million traininable parameters.")
+
 col1, col2 = st.columns(2)
 
 if 'text_input' not in st.session_state:
@@ -33,7 +35,7 @@ if 'warnings' not in st.session_state:
 with col1:
  with st.container(border=True):
  st.markdown("Input :clipboard:")
- sample_sentence = "Vhana vhane vha kha ḓi bva u bebwa vha kha khombo ya u <mask> nga Listeriosis."
+ sample_sentence = "Rabulasi wa <mask> u khou bvelela nga u lima"
  
  text_input = st.text_area(
  "Enter sentences with <mask> token:",