jasonsurya0/BART_Text_Summarizer

1import streamlit as st 2from transformers import pipeline3 4def summarizeText(summarizer, txt):5    summarized_text = summarizer(txt, max_length=100)[0]["summary_text"]6    return summarized_text7 8@st.cache_resource9def load_summarizer():10    return pipeline("summarization", model = "jasonsurya0/BART_TWELVE")11 12def main():13    st.set_page_config(page_title="Automatic Text Summarizer With BART")14    # BART MODEL DEVELOPED 15    summarizer = load_summarizer()16    # ----HEADER17    st.subheader("Text Summarizer Built With BART")18 19    #-----Text Area20    txt = st.text_area('Input Text to Summarize', '''21        Amanda: I baked cookies. Do you want some? 22        Jerry: Sure! 23        Amanda: I'll bring you tomorrow :-) 24        ''', height = 180)25 26    if st.button('Summarize'):27        # st.write("TEST")28        st.text_area('Summarized Text', summarizeText(summarizer, txt), height = 140)29 30if __name__=="__main__":31    main()