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()