Update appStore/multiapp.py

appStore/multiapp.py

@@ -16,6 +16,17 @@ logger = logging.getLogger(__name__)
 if 'file' not in st.session_state:
     st.session_state['pipeline'] = None
     
+#Haystack Components
+@st.cache(hash_funcs={"builtins.SwigPyObject": lambda _: None},allow_output_mutation=True)
+
+def start_haystack(documents_processed):
+    document_store = InMemoryDocumentStore()
+    document_store.write_documents(documents_processed)
+    retriever = TfidfRetriever(document_store=document_store)
+    reader = FARMReader(model_name_or_path="deepset/tinyroberta-squad2", use_gpu=True) #deepset/roberta-base-squad2
+    pipeline = ExtractiveQAPipeline(reader, retriever)
+    return pipeline
+    
 class MultiApp:
     """
     Framework for combining multiple streamlit applications.