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