Update app.py

app.py

@@ -28,9 +28,9 @@ def question_answering(question):
     print(question)
     query_sentence = [question]
     query_embedding = st.session_state.model_embedding.encode(query_sentence)
-    k = 5
+    k = 20
     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 = 512) for i in range(k)]
+    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'])
     print(best_answer['answer'])
     if best_answer['score'] > 0.5:
@@ -47,7 +47,7 @@ for message in st.session_state.messages:
 
 def clean_answer(s):
     # Sử dụng regex để loại bỏ tất cả các ký tự đặc biệt ở cuối chuỗi
-    return re.sub(r'[^a-zA-Z0-9]+$', '', s)
+    return re.sub(r'[^aAàÀảẢáÁạẠăĂằẰẳẲẵẴắẮặẶâÂầẦẩẨẫẪấẤậẬbBcCdDđĐeEèÈẻẺẽẼéÉẹẸêÊềỀểỂễỄếẾệỆfFgGhHiIìÌỉỈĩĨíÍịỊjJkKlLmMnNoOòÒỏỎõÕóÓọỌôÔồỒổỔỗỖốỐộỘơƠờỜởỞỡỠớỚợỢpPqQrRsStTuUùÙủỦũŨúÚụỤưƯừỪửỬữỮứỨựỰvVwWxXyYỳỲỷỶỹỸýÝỵỴzZ0-9]+$', '', s)
     
 if prompt := st.chat_input("What is up?"):
         st.session_state.messages.append({"role": "user", "content": prompt})