Admit failure when answer is not known

test_answers.py

@@ -67,6 +67,12 @@ def test_home():
     assert a == "Villafranca di Verona"
 
 
+def test_address():
+    q = "Qual è l'indirizzo di casa?"
+    a = get_answer(q, context, nlp_qa)
+    assert "Vittorio Emanuele II" in a
+
+
 def test_history():
     q = "Cosa mi è successo?"
     a = get_answer(q, context, nlp_qa)

utils.py

@@ -6,11 +6,14 @@ import torch
 from transformers import Pipeline, pipeline
 
 
-def get_answer(input, context, engine):
+def get_answer(input, context, engine, threshold=0.4):
 
     answer = engine({"question": input, "context": context})
 
-    return answer["answer"]
+    if answer["score"] > threshold:
+        return answer["answer"]
+    else:
+        return "Non lo so, prova con un'altra domanda!"
 
 
 @st.cache