Linhz commited on
Commit
9d0e5e0
·
verified ·
1 Parent(s): b70c076

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -48,9 +48,11 @@ def question_answering(question):
48
  answer = [question_answerer(question=query_sentence[0], context=articles[I[0][i]], max_answer_len = 512) for i in range(k)]
49
  best_answer = max(answer, key=lambda x: x['score'])
50
  print(best_answer)
51
- if best_answer['score'] > 0.5:
52
  return best_answer['answer']
53
- return f"Tôi không chắc lắm nhưng có lẽ câu trả lời là: \n{best_answer['answer']}"
 
 
54
 
55
  # if "messages" not in st.session_state:
56
  # st.session_state.messages = []
 
48
  answer = [question_answerer(question=query_sentence[0], context=articles[I[0][i]], max_answer_len = 512) for i in range(k)]
49
  best_answer = max(answer, key=lambda x: x['score'])
50
  print(best_answer)
51
+ if best_answer['score'] > 0.7:
52
  return best_answer['answer']
53
+ else if best_answer['score'] > 0.3:
54
+ return f"Tôi không chắc lắm nhưng có lẽ câu trả lời là: \n{best_answer['answer']}"
55
+ return f"Xin lỗi tôi không biết câu trả lời cho câu hỏi này, vui lòng hỏi lại câu hỏi khác"
56
 
57
  # if "messages" not in st.session_state:
58
  # st.session_state.messages = []