Tuana commited on
Commit
07607d7
1 Parent(s): 95e1f5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ def start_haystack():
16
  split_respect_sentence_boundary=True,
17
  split_overlap=0
18
  )
19
- summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-xsum")
20
  return document_store, summarizer, preprocessor
21
 
22
 
@@ -34,7 +34,7 @@ def pdf_to_document_store(pdf_files):
34
 
35
  def summarize(files):
36
  pdf_to_document_store(files)
37
- summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
38
  st.write(summary)
39
 
40
  document_store, summarizer, preprocessor = start_haystack()
 
16
  split_respect_sentence_boundary=True,
17
  split_overlap=0
18
  )
19
+ summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-newsroom")
20
  return document_store, summarizer, preprocessor
21
 
22
 
 
34
 
35
  def summarize(files):
36
  pdf_to_document_store(files)
37
+ summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=False)
38
  st.write(summary)
39
 
40
  document_store, summarizer, preprocessor = start_haystack()