Team Ai
Apppublic

Nn00/sample-classification

sourceHugging Facecc-by-nc-4.0updated 4y agoView on Hugging Face
0likes
app.py18 linesDownload Raw Back to root
1import streamlit as st2from transformers import pipeline3 4classifier = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")5def main():6    #Token Sample7    st.title("Summarize Classification ⋆·˚ ༘ *")8 9    with st.form("text_field"):10        text = st.text_area('enter some text:')11        # clicked==True only when the button is clicked12        clicked = st.form_submit_button("Done!")13        if clicked:14          results = classifier([text])15          st.json(results)16 17if __name__ == "__main__":18    main()