From 70c8065086e454762fa76fad6137653c388a604c Mon Sep 17 00:00:00 2001 From: teaxu Date: Sat, 4 Apr 2020 16:03:01 +0800 Subject: [PATCH] fix bug --- project/main.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/project/main.py b/project/main.py index c312f52..4911c38 100644 --- a/project/main.py +++ b/project/main.py @@ -3,11 +3,13 @@ import json from retrieval import information_retrieval -from qa import QaModule, print_answers_in_file +from caireCovid import QaModule, print_answers_in_file all_results, data_for_qa = information_retrieval("question_generation/task1_question.json") -qa_model = QaModule(["mrqa", "biobert"]) +qa_model = QaModule(["mrqa", "biobert"], ["/home/xuyan/mrqa/xlnet-qa/experiment/multiqa-1e-5-tpu/1564469515", "/home/xuyan/kaggle/bioasq-biobert/model/1585470591"], \ + "./mrqa/model/spiece.model", "./biobert/model/bert_config.json", "./biobert/model/vocab.txt" +) answers = qa_model.getAnswers(data_for_qa)