Spaces:

iloncka
/

QA-project-ITMO-Science-cases

Runtime error

iloncka commited on Feb 2, 2023

Commit

65bddbe

1 Parent(s): 40769e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ def get_doc_reg_id(result):
 document_store = InMemoryDocumentStore() # use_bm25=True
 load_and_write_data(document_store)
 retriever = TfidfRetriever(document_store=document_store)
-reader = FARMReader(model_name_or_path="mrm8488/RuPERTa-base-finetuned-squadv1",
                     use_gpu=False,
                     num_processes=1)
 pipeline = ExtractiveQAPipeline(reader, retriever)

 document_store = InMemoryDocumentStore() # use_bm25=True
 load_and_write_data(document_store)
 retriever = TfidfRetriever(document_store=document_store)
+reader = FARMReader(model_name_or_path="DeepPavlov/rubert-base-cased-sentence",
                     use_gpu=False,
                     num_processes=1)
 pipeline = ExtractiveQAPipeline(reader, retriever)