Update app.py

app.py

@@ -4,7 +4,12 @@ import streamlit as st
 unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
 
 def main():
-  st.title('EN mask filling')
-  st.write('Please insert english sentence with [MASK] to let model fill a mask')
-  st.write('For example, I [MASK] dog for a walk')
-  
+  st.title('EN sentence mask filling')
+  st.markdown('Mask a word with [MASK]. For example, **I [MASK] dog for a walk**')
+  masked = st.text_input('Masked sentence','Please insert a sentence')
+  
+  output = unmasker(masked)
+  
+ 
+ if __name__ == "__main__":
+    main()