Update app.py

app.py

@@ -7,8 +7,6 @@ from transformers import pipeline
 import torch
 
 
-
-
 st.set_page_config(page_title = "Vietnamese Legal Question Answering System", page_icon= "🐧", layout="centered", initial_sidebar_state="collapsed")
 
 st.markdown(
@@ -37,7 +35,7 @@ def question_answering(question):
     print(question)
     query_sentence = [question]
     query_embedding = st.session_state.model_embedding.encode(query_sentence)
-    k = 100
+    k =500
     D, I = index_loaded.search(query_embedding.astype('float32'), k)  # D is distances, I is indices
     answer = [question_answerer(question=query_sentence[0], context=articles[I[0][i]], max_answer_len = 256) for i in range(k)]
     best_answer = max(answer, key=lambda x: x['score'])