demo-test00 / app.py
bluebear27's picture
Update app.py
25a7d76
raw
history blame
207 Bytes
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)