Linhz commited on
Commit
6ae05ec
·
verified ·
1 Parent(s): 50d9e8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -37,9 +37,11 @@ def question_answering(question):
37
  query_embedding = st.session_state.model_embedding.encode(query_sentence)
38
  k = 100
39
  D, I = index_loaded.search(query_embedding.astype('float32'), k) # D is distances, I is indices
40
- answer = [question_answerer(question=query_sentence[0], context=articles[I[0][i]], max_answer_len = 512) for i in range(k)]
41
- best_answer = max(answer, key=lambda x: x['score'])
42
- print(best_answer['answer'])
 
 
43
  if best_answer['score'] > 0.5:
44
  return best_answer['answer']
45
  return f"Tôi không chắc lắm nhưng có lẽ câu trả lời là: \n{best_answer['answer']}"
 
37
  query_embedding = st.session_state.model_embedding.encode(query_sentence)
38
  k = 100
39
  D, I = index_loaded.search(query_embedding.astype('float32'), k) # D is distances, I is indices
40
+ # answer = [question_answerer(question=query_sentence[0], context=articles[I[0][i]], max_answer_len = 512) for i in range(k)]
41
+ # best_answer = max(answer, key=lambda x: x['score'])
42
+ context = [articles[I[0][i]] for i in range(k)]
43
+ best_answer = question_answerer(question=query_sentence[0], context= context, max_answer_len = 512)
44
+ print(best_answer)
45
  if best_answer['score'] > 0.5:
46
  return best_answer['answer']
47
  return f"Tôi không chắc lắm nhưng có lẽ câu trả lời là: \n{best_answer['answer']}"