Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from haystack.document_stores import InMemoryDocumentStore
-from haystack.nodes import TransformersSummarizer, PreProcessor, PDFToTextConverter, TfidfRetriever
+from haystack.nodes import TransformersSummarizer, PreProcessor, PDFToTextConverter
 from haystack.schema import Document
 import logging
 import base64
@@ -17,7 +17,7 @@ def start_haystack():
         split_length=200,
         split_respect_sentence_boundary=True,
     )
-    summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-newsroom")
+    summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-xsum")
     return document_store, summarizer, preprocessor
 
 
@@ -36,7 +36,7 @@ def pdf_to_document_store(pdf_files):
 
 def summarize(files):
     pdf_to_document_store(files)
-    summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=False)
+    summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
     st.write(summary)
 
 document_store, summarizer, preprocessor = start_haystack()