Update app.py

app.py

@@ -48,7 +48,7 @@ list_nodes = [node.text for node in nodes]
 print(os.getcwd())
 
 # if os.path.exists(DOC_PATH):
-RAG = RAGPretrainedModel.from_index(DOC_PATH)
+RAG = RAGPretrainedModel.from_index(DOC_PATH, n_gpu=None)
 # else:
 #   RAG = RAGPretrainedModel.from_pretrained("AdrienB134/ColBERTv2.0-spanish-mmarcoES")
 #   my_documents = list_nodes
@@ -73,7 +73,7 @@ def reformat_rag(results_rag):
 
 def chat_stream_completion(message, history):
 
-    context = reformat_rag(RAG.search(message, k=1))
+    context = reformat_rag(RAG.search(message, DOC_PATH, k=2))
     context = " \n ".join(context)
 
     full_prompt = prompt.format(context,message,"")