Spaces:
Build error
Build error
Update app.py
Browse files
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-
|
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=
|
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()
|