Spaces:
Build error
Build error
batch size
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def start_haystack():
|
|
17 |
split_length=200,
|
18 |
split_respect_sentence_boundary=True,
|
19 |
)
|
20 |
-
summarizer = TransformersSummarizer(model_name_or_path="facebook/bart-large-cnn"
|
21 |
return document_store, summarizer, preprocessor
|
22 |
|
23 |
|
@@ -34,7 +34,7 @@ def pdf_to_document_store(pdf_file):
|
|
34 |
|
35 |
def summarize(file):
|
36 |
pdf_to_document_store(file)
|
37 |
-
summaries = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
|
38 |
st.write('Summary')
|
39 |
for summary in summaries:
|
40 |
st.write(summary.content)
|
|
|
17 |
split_length=200,
|
18 |
split_respect_sentence_boundary=True,
|
19 |
)
|
20 |
+
summarizer = TransformersSummarizer(model_name_or_path="facebook/bart-large-cnn")
|
21 |
return document_store, summarizer, preprocessor
|
22 |
|
23 |
|
|
|
34 |
|
35 |
def summarize(file):
|
36 |
pdf_to_document_store(file)
|
37 |
+
summaries = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True, batch_size=5))
|
38 |
st.write('Summary')
|
39 |
for summary in summaries:
|
40 |
st.write(summary.content)
|