Nn00/sample-classification
0
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()