shripadbhat commited on
Commit
bd41160
1 Parent(s): 5ad2a61

Update question_answering.py

Browse files
Files changed (1) hide show
  1. question_answering.py +1 -1
question_answering.py CHANGED
@@ -14,7 +14,7 @@ class QuestionAnswering:
14
  self.passage_retreival_model = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
15
  self.qa_model = pipeline("question-answering",'a-ware/bart-squadv2')
16
 
17
- def fetch_answers(question, document):
18
  document_paragraphs = document.splitlines()
19
  query_paragraph_list = [(question, para) for para in document_paragraphs if len(para.strip()) > 0 ]
20
 
 
14
  self.passage_retreival_model = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
15
  self.qa_model = pipeline("question-answering",'a-ware/bart-squadv2')
16
 
17
+ def fetch_answers(self, question, document):
18
  document_paragraphs = document.splitlines()
19
  query_paragraph_list = [(question, para) for para in document_paragraphs if len(para.strip()) > 0 ]
20