app.py

from transformers import pipeline
import streamlit as st

unmasker = pipeline('fill-mask', model='distilbert-base-uncased')

def main():
  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()