PaiRin/Text_Token_Classification

1import streamlit as st2from transformers import pipeline3 4classifier = pipeline("token-classification", model="distilbert-base-uncased")5def main():6    st.title("Text Token Classification")7 8    with st.form("text_field"):9        text = st.text_area('type some text:')10        # clicked==True only when the button is clicked11        clicked = st.form_submit_button("Submit")12        if clicked:13          results = classifier([text])14          st.json(results)15 16if __name__ == "__main__":17    main()