cache haystack setup

app.py

@@ -4,17 +4,20 @@ from haystack.nodes import TransformersSummarizer, PreProcessor, PDFToTextConver
 from haystack.schema import Document
 import logging
 
-document_store = InMemoryDocumentStore()
-preprocessor = PreProcessor(
-    clean_empty_lines=True,
-    clean_whitespace=True,
-    clean_header_footer=True,
-    split_by="word",
-    split_length=100,
-    split_respect_sentence_boundary=True,
-    split_overlap=3
-)
-summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-xsum")
+@st.cache(hash_funcs={"builtins.SwigPyObject": lambda _: None},allow_output_mutation=True)
+def start_haystack():
+    document_store = InMemoryDocumentStore()
+    preprocessor = PreProcessor(
+        clean_empty_lines=True,
+        clean_whitespace=True,
+        clean_header_footer=True,
+        split_by="word",
+        split_length=100,
+        split_respect_sentence_boundary=True,
+        split_overlap=3
+    )
+    summarizer = TransformersSummarizer(model_name_or_path="google/pegasus-xsum")
+    return document_store, summarizer, preprocessor
 
 
 def pdf_to_document_store(pdf_files):
@@ -33,6 +36,7 @@ def summarize(files):
     summary = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
     st.write(summary)
 
+document_store, summarizer, preprocessor = start_haystack()
 
 uploaded_files = st.file_uploader("Choose PDF files", accept_multiple_files=True)