Spaces:
Running
Running
File size: 646 Bytes
df4609d bd6e14e 3088252 e50c5bd 1d90c39 df4609d 46c7402 1d90c39 ac48c85 5b07eb9 1d90c39 713593f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import streamlit as st
from transformers import pipeline
st.set_page_config(page_title="Text Summarisation", page_icon="🤖", layout="wide", initial_sidebar_state="expanded",)
st.title("Text Summarisation")
pipe = pipeline(task="summarization", model="facebook/bart-large-cnn")
text_value = st.text_area("Summarise the following")
if text_value:
out = pipe(text_value, max_length=int(len(text_value)/2), min_length=int(len(text_value)/3), do_sample=False)
if len(out) >= 1:
st.write(out[0]['summary_text'])
st.download_button("summary download", out[0]['summary_text'], file_name="summersation.txt", mime="text/plain")
|