Update app.py

app.py

@@ -74,18 +74,21 @@ model._is_chat_session_activated = False
 
 max_new_tokens = 2048
 
-
-
-
-
-
-
-def generater(message, history, temperature, top_p, top_k):
+def generater(message, history):
     prompt = "<s>"
     for user_message, assistant_message in history:
         prompt += model.config["promptTemplate"].format(user_message)
-        prompt += assistant_message + "</s>"
+        prompt += assistant_message +  "</s>"
     prompt += model.config["promptTemplate"].format(message)
+
+    question_embeddings = np.array([get_text_embedding(message)])
+
+    D, I = index.search(question_embeddings, k=3) # distance, index
+    retrieved_chunk = [chunks[i] for i in I.tolist()[0]]
+
+    prompt += "Informations de contexte ci-dessous:"
+    prompt += [i for i in retrieved_chunk]
+    
     outputs = []    
     for token in model.generate(prompt=prompt, temp=0.5, top_k = 40, top_p = 1, max_tokens = max_new_tokens, streaming=True):
         outputs.append(token)
@@ -97,7 +100,7 @@ def vote(data: gr.LikeData):
     else:
         return
 
-chatbot = gr.Chatbot(avatar_images=('resourse/user-icon.png', 'resourse/chatbot-icon.png'),bubble_full_width = False)
+chatbot = gr.Chatbot(avatar_images=('./resourse/user-icon.png', './resourse/chatbot-icon.png'),bubble_full_width = False)
 """
 additional_inputs=[
     gr.Slider(