Update app.py

app.py

@@ -38,7 +38,7 @@ def question_answering(question):
     print(question)
     query_sentence = [question]
     query_embedding = st.session_state.model_embedding.encode(query_sentence)
-    k = 500
+    k = 1
     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)]
     best_answer = max(answer, key=lambda x: x['score'])
@@ -76,11 +76,11 @@ for message in st.session_state.messages:
     if message['role'] == 'assistant':
         avatar_class = "assistant-avatar"
         message_class = "assistant-message"
-        avatar = './static/AI.png'
+        avatar = './app/static/static_ai.png'
     else:
         avatar_class = "user-avatar"
         message_class = "user-message"
-        avatar = './static/human.jpg'
+        avatar = './app/static/static_human.png'
     st.markdown(f"""
     <div class="{message_class}">
         <img src="{avatar}" class="{avatar_class}" />