shripadbhat commited on
Commit
8a010b4
1 Parent(s): 3efe6ba

Update question_answering.py

Browse files
Files changed (1) hide show
  1. question_answering.py +2 -2
question_answering.py CHANGED
@@ -46,8 +46,8 @@ class QuestionAnswering:
46
  attention_mask = encoded_input.attention_mask)
47
  output_answer = self.tokenizer.decode(output[0], skip_special_tokens=True)
48
 
49
- result_str = "# ANSWER "+str(count)+": "+ output_answer +"\n"
50
- result_str = result_str + "REFERENCE: "+ evidence_sentence + "\n\n"
51
  top_5_query_paragraph_answer_list += result_str
52
  count+=1
53
 
 
46
  attention_mask = encoded_input.attention_mask)
47
  output_answer = self.tokenizer.decode(output[0], skip_special_tokens=True)
48
 
49
+ result_str = ""+str(count)+": "+ output_answer +"\n"
50
+ result_str = result_str + " "+ evidence_sentence + "\n\n"
51
  top_5_query_paragraph_answer_list += result_str
52
  count+=1
53