Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import streamlit as st
|
2 |
from haystack.document_stores import InMemoryDocumentStore
|
3 |
-
from haystack.nodes import TransformersSummarizer, PreProcessor, PDFToTextConverter
|
4 |
from haystack.schema import Document
|
5 |
import logging
|
6 |
import base64
|
@@ -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="google/pegasus-
|
21 |
return document_store, summarizer, preprocessor
|
22 |
|
23 |
|
@@ -36,7 +36,7 @@ def pdf_to_document_store(pdf_files):
|
|
36 |
|
37 |
def summarize(files):
|
38 |
pdf_to_document_store(files)
|
39 |
-
summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=
|
40 |
st.write(summary)
|
41 |
|
42 |
document_store, summarizer, preprocessor = start_haystack()
|
|
|
1 |
import streamlit as st
|
2 |
from haystack.document_stores import InMemoryDocumentStore
|
3 |
+
from haystack.nodes import TransformersSummarizer, PreProcessor, PDFToTextConverter
|
4 |
from haystack.schema import Document
|
5 |
import logging
|
6 |
import base64
|
|
|
17 |
split_length=200,
|
18 |
split_respect_sentence_boundary=True,
|
19 |
)
|
20 |
+
summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-xsum")
|
21 |
return document_store, summarizer, preprocessor
|
22 |
|
23 |
|
|
|
36 |
|
37 |
def summarize(files):
|
38 |
pdf_to_document_store(files)
|
39 |
+
summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
|
40 |
st.write(summary)
|
41 |
|
42 |
document_store, summarizer, preprocessor = start_haystack()
|