UnarineLeo
commited on
Commit
•
e976ec9
1
Parent(s):
7a621b0
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def replace_mask(sentence, predicted_word):
|
|
27 |
st.title("Fill Mask for Multiple Languages | Zabantu-Bantu-250m")
|
28 |
st.write("This app predicts the missing word for sentences in Zulu, Tshivenda, Sepedi, Tswana, and Tsonga using a Zabantu BERT model.")
|
29 |
|
30 |
-
user_sentence = st.text_input("Enter your own sentence with a masked word (use '____'):", "\n".join(sample_sentences))
|
31 |
|
32 |
if st.button("Submit"):
|
33 |
user_masked_sentence = user_sentence.replace('____', unmasker.tokenizer.mask_token)
|
|
|
27 |
st.title("Fill Mask for Multiple Languages | Zabantu-Bantu-250m")
|
28 |
st.write("This app predicts the missing word for sentences in Zulu, Tshivenda, Sepedi, Tswana, and Tsonga using a Zabantu BERT model.")
|
29 |
|
30 |
+
user_sentence = st.text_input("Enter your own sentence with a masked word (use '____'):", "\n".join(sample_sentences.values()))
|
31 |
|
32 |
if st.button("Submit"):
|
33 |
user_masked_sentence = user_sentence.replace('____', unmasker.tokenizer.mask_token)
|