import streamlit as st
from transformers import pipeline

st.title('BERT Summarizer')
uploaded_file = st.file_uploader("Choose a text file", type="txt")

if uploaded_file is not None:
    data = uploaded_file.read().decode('utf-8')
    data = data.replace('\n', '')
    data = data.replace("\ufeff", "")

    summarizer = pipeline("summarization")
    summarized = summarizer(data[:1024]) # BERT has a max token limit
    st.write(summarized[0]['summary_text'])