Update app.py

app.py

@@ -8,10 +8,10 @@ from bs4 import BeautifulSoup
 import re
 
 benefits = [
-  {"benefitName": "Universal Credit", "coreName": "Overview", "link": "https://www.gov.uk/universal-credit/"},
-  {"benefitName": "Universal Credit", "coreName": "Eligibility", "link": "https://www.gov.uk/universal-credit/eligibility"},
-  {"benefitName": "Universal Credit", "coreName": "how much can I get​", "link": "https://www.gov.uk/universal-credit/what-youll-get,https://www.gov.uk/universal-credit/how-youre-paid"},
-  {"benefitName": "Universal Credit", "coreName": "how to apply/claim", "link": "https://www.gov.uk/universal-credit/how-to-claim"},
+  {"benefitName": "Universal Credit", "coreName": "what is this benefit", "link": "https://www.gov.uk/universal-credit/"},
+  {"benefitName": "Universal Credit", "coreName": "who can apply", "link": "https://www.gov.uk/universal-credit/eligibility"},
+  {"benefitName": "Universal Credit", "coreName": "how much can I get", "link": "https://www.gov.uk/universal-credit/what-youll-get,https://www.gov.uk/universal-credit/how-youre-paid"},
+  {"benefitName": "Universal Credit", "coreName": "How to apply", "link": "https://www.gov.uk/universal-credit/how-to-claim"}
 ]
 
 
@@ -93,7 +93,21 @@ core4Classes = list(set(list(map(lambda x: x['coreName'], benefits))))
 # contexts
 benefitsClasses, core4Classes
 
-#question_answerer = pipeline("question-answering")
+question_answerer = pipeline("question-answering")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+
+coreName = 'how much can I get'
+def testQA(question):
+
+  predictedBenefit = "Universal Credit"
+  predictedCore = coreName
+
+  time = datetime.now()
+  context = list(filter(lambda x: x['benefitName']==predictedBenefit and x['coreName']==predictedCore, benefits))[0]
+  answer = question_answerer(question = question, context = context['context'])
+  time3 = (datetime.now() - time).total_seconds()
+
+  return answer
+  
+iface = gr.Interface(fn=testQA, inputs="text", outputs="text")
 iface.launch()