Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
unmasker = pipeline('fill-mask', model='dsfsi/zabantu-ven-
|
5 |
|
6 |
def fill_mask(sentences):
|
7 |
results = {}
|
@@ -10,8 +10,7 @@ def fill_mask(sentences):
|
|
10 |
results[sentence] = unmasked
|
11 |
return results
|
12 |
|
13 |
-
st.title("Fill Mask | Zabantu-
|
14 |
-
st.write("This app predicts the missing word in a sentence using a BERT model.")
|
15 |
|
16 |
col1, col2 = st.columns(2)
|
17 |
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
unmasker = pipeline('fill-mask', model='dsfsi/zabantu-sot-ven-170m')
|
5 |
|
6 |
def fill_mask(sentences):
|
7 |
results = {}
|
|
|
10 |
results[sentence] = unmasked
|
11 |
return results
|
12 |
|
13 |
+
st.title("Fill Mask | Zabantu-sot-ven-170m")
|
|
|
14 |
|
15 |
col1, col2 = st.columns(2)
|
16 |
|