Spaces:

peter2000
/

policy_test

Runtime error

peter2000 commited on Sep 28, 2022

Commit

9640adc

•

1 Parent(s): f2425b0

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="multi-qa-MiniLM-L6-cos-v1", 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="deepset/roberta-base-squad2", use_gpu=True)
     pipeline = ExtractiveQAPipeline(reader, retriever)
     return pipeline