Update app.py

app.py

@@ -46,6 +46,7 @@ MODELS = [
     "mistralai/Mathstral-7B-v0.1",
     "meta-llama/Meta-Llama-3.1-8B-Instruct",
     "meta-llama/Meta-Llama-3.1-70B-Instruct",
+    "mattshumer/Reflection-Llama-3.1-70B",
     "gemini-1.5-flash",
     "duckduckgo/gpt-4o-mini",
     "duckduckgo/claude-3-haiku",
@@ -498,7 +499,7 @@ def get_response_from_llama(query, model, selected_docs, file_type, num_calls=1,
         # Get the context from selected Excel documents
         embed = get_embeddings()
         office_database = FAISS.load_local("office_faiss_database", embed, allow_dangerous_deserialization=True)
-        retriever = office_database.as_retriever(search_kwargs={"k": 10})
+        retriever = office_database.as_retriever(search_kwargs={"k": 20})
         relevant_docs = retriever.get_relevant_documents(query)
         context = "\n".join([doc.page_content for doc in relevant_docs if doc.metadata["source"] in selected_docs])