walter1 commited on
Commit
cbf83a0
1 Parent(s): 551bca9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -104,7 +104,7 @@ def testQA(question):
104
 
105
  time = datetime.now()
106
  context = list(filter(lambda x: x['benefitName']==predictedBenefit and x['coreName']==predictedCore, benefits))[0]
107
- answer = question_answerer(question = question, context = context['context'])
108
  time3 = (datetime.now() - time).total_seconds()
109
 
110
  return answer
104
 
105
  time = datetime.now()
106
  context = list(filter(lambda x: x['benefitName']==predictedBenefit and x['coreName']==predictedCore, benefits))[0]
107
+ answer = question_answerer(question = question, context = context['context'])['answer']
108
  time3 = (datetime.now() - time).total_seconds()
109
 
110
  return answer