jchen8000 commited on
Commit
1f5b72f
·
verified ·
1 Parent(s): f16ca94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def generate_response(query, history, model, temperature, max_tokens, top_p, see
89
  if seed == 0:
90
  seed = random.randint(1, 100000)
91
 
92
- retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 8})
93
  llm = ChatGroq(groq_api_key=os.environ.get("GROQ_API_KEY"), model=model)
94
  custom_rag_prompt = PromptTemplate.from_template(template)
95
 
 
89
  if seed == 0:
90
  seed = random.randint(1, 100000)
91
 
92
+ retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 16})
93
  llm = ChatGroq(groq_api_key=os.environ.get("GROQ_API_KEY"), model=model)
94
  custom_rag_prompt = PromptTemplate.from_template(template)
95