import streamlit as st
import os
from transformers import pipeline


fill_masker = pipeline(model="bert-base-uncased")
# fill_masker("This is a simple [MASK].")
 
if query := st.chat_input("filler: "):
     ans = fill_masker(f"{query}[MASK]")
     with st.chat_message("User"):
         st.write(ans)