Update app.py

app.py

@@ -17,7 +17,7 @@ def start_haystack():
         split_length=200,
         split_respect_sentence_boundary=True,
     )
-    summarizer = TransformersSummarizer(model_name_or_path="google/bigbird-pegasus-large-arxiv")
+    summarizer = TransformersSummarizer(model_name_or_path="facebook/bart-large-cnn")
     return document_store, summarizer, preprocessor
 
 
@@ -36,14 +36,26 @@ def summarize(file):
     pdf_to_document_store(file)
     st.write('Number of documents', document_store.get_document_count())
     summaries = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
-    st.write('Summary')
-    for summary in summaries:
-        st.write(summary.content)
+    return summaries
+
+def set_state_if_absent(key, value):
+    if key not in st.session_state:
+        st.session_state[key] = value
         
+set_state_if_absent("summaries", None)        
 document_store, summarizer, preprocessor = start_haystack()
 
 uploaded_file = st.file_uploader("Choose a PDF file", accept_multiple_files=False)
 
 if uploaded_file is not None:
     if st.button('Summarize Document'):
-        summarize(uploaded_file)
+        with st.spinner("📚    Please wait while we produce a summary..."):
+            try:
+                st. session_state.summaries = summarize(uploaded_file)
+            except Exception as e:
+                logging.exception(e)
+ 
+if st.session_state.summaries:
+    st.write('## Summary')
+    for count, summary in enumerate(st.session_state.summaries):
+        st.write(summary.content)