Update appStore/keyword_search.py

appStore/keyword_search.py

@@ -21,10 +21,8 @@ import numpy as np
 #Haystack Components
 @st.cache(hash_funcs={"builtins.SwigPyObject": lambda _: None},allow_output_mutation=True)
 
-def start_haystack(temp.name, file):
+def start_haystack(documents_processed):
     document_store = InMemoryDocumentStore()
-    documents = pre.load_document(temp.name, file)
-    documents_processed = pre.preprocessing(documents)
     document_store.write_documents(documents_processed)
     retriever = TfidfRetriever(document_store=document_store)
     reader = FARMReader(model_name_or_path="deepset/roberta-base-squad2-distilled", use_gpu=True)
@@ -83,7 +81,9 @@ def app():
                 st.write("Filename: ", file.name)
                 
                 # load document
-                pipeline = start_haystack(temp.name, file)
+                documents = pre.load_document(temp.name, file)
+                documents_processed = pre.preprocessing(documents)
+                pipeline = start_haystack(documents_processed)
                 #docs = pre.load_document(temp.name, file)
 
                 # preprocess document