Spaces:
Runtime error
Runtime error
import streamlit as st | |
from transformers import pipeline | |
def summarizeText(summarizer, txt): | |
summarized_text = summarizer(txt, max_length=100)[0]["summary_text"] | |
return summarized_text | |
def load_summarizer(): | |
return pipeline("summarization", model = "jasonsurya0/BART_TWELVE") | |
def main(): | |
st.set_page_config(page_title="Automatic Text Summarizer With BART") | |
# BART MODEL DEVELOPED | |
summarizer = load_summarizer() | |
# ----HEADER | |
st.subheader("Text Summarizer Built With BART") | |
#-----Text Area | |
txt = st.text_area('Input Text to Summarize', ''' | |
Amanda: I baked cookies. Do you want some? | |
Jerry: Sure! | |
Amanda: I'll bring you tomorrow :-) | |
''', height = 180) | |
if st.button('Summarize'): | |
# st.write("TEST") | |
st.text_area('Summarized Text', summarizeText(summarizer, txt), height = 140) | |
if __name__=="__main__": | |
main() |