Update app.py

app.py

@@ -26,17 +26,22 @@ config['device'] = 'cpu' # mps
 recognitor = Predictor(config)
 model_name = "microsoft/xdoc-base-squad2.0"
 nlp = pipeline('question-answering', model=model_name)
-def query(doc, candidates):
-    questions = candidates.split(", ")
+
+def query(doc, labels):
+    questions = labels.split(", ")
     result={}
     for question in questions:
         QA_input = {
                     'question': question,
                     'context': doc
                 }
-            
+        print(QA_input)
         res= nlp(QA_input)
-        value = res['answer']
+        print(res)
+        if res['score']>0.5:
+            value = res['answer']
+        else:
+            value=''
         result[question]=value
     return result