Spaces:

peter2000
/

policy_test

Runtime error

peter2000 commited on Sep 28, 2022

Commit

48bb07e

•

1 Parent(s): d5e831b

Update appStore/keyword_search.py

Files changed (1) hide show

appStore/keyword_search.py CHANGED Viewed

@@ -32,7 +32,7 @@ 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/roberta-base-squad2", use_gpu=True)
     pipeline = ExtractiveQAPipeline(reader, retriever)
     return pipeline

     document_store = InMemoryDocumentStore()
     document_store.write_documents(documents_processed)
     retriever = TfidfRetriever(document_store=document_store)
+    reader = FARMReader(model_name_or_path="allenai/longformer-large-4096-finetuned-triviaqa", use_gpu=True) #deepset/roberta-base-squad2
     pipeline = ExtractiveQAPipeline(reader, retriever)
     return pipeline