Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
-from summarizer import Summarizer
-from pprint import pprint
+from transformers import pipeline
 
 st.title('BERT Summarizer')
 uploaded_file = st.file_uploader("Choose a text file", type="txt")
@@ -10,7 +9,6 @@ if uploaded_file is not None:
     data = data.replace('\n', '')
     data = data.replace("\ufeff", "")
 
-    model = Summarizer()
-    result = model(data, min_length=60, max_length=500)
-    summarized_text = ''.join(result)
-    st.write(summarized_text)
+    summarizer = pipeline("summarization")
+    summarized = summarizer(data[:1024]) # BERT has a max token limit
+    st.write(summarized[0]['summary_text'])