nakere424 commited on
Commit
7337250
·
1 Parent(s): f76cd25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -4,11 +4,11 @@ import streamlit as st
4
  unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
5
 
6
  def main():
7
- st.title('EN sentence mask filling')
8
- st.markdown('Mask a word with [MASK]. For example, **I [MASK] dog for a walk**')
9
- masked = st.text_input('Masked sentence','Please insert a sentence')
10
 
11
- output = unmasker(masked)
12
 
13
  if __name__ == "__main__":
14
- main()
 
4
  unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
5
 
6
  def main():
7
+ st.title('EN sentence mask filling')
8
+ st.markdown('Mask a word with [MASK]. For example, **I [MASK] dog for a walk**')
9
+ masked = st.text_input('Masked sentence','Please insert a sentence')
10
 
11
+ output = unmasker(masked)
12
 
13
  if __name__ == "__main__":
14
+ main()