import streamlit as st from transformers import pipeline pipe = pipeline("fill mask", model = "sarkerlab/SocBERT-base") text = st.text_area('enter some text') if text: out = pipe(text) st.json(out)