UnarineLeo commited on
Commit
be35644
·
verified ·
1 Parent(s): 51fb456

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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-120m')
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-Bantu-250m")
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