Jasur05 commited on
Commit
6e25b8d
Β·
1 Parent(s): dbd5bdd
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ def generate_agent_answer(context: str, question: str) -> str:
105
  return response.text.strip()
106
 
107
  def rag_answer(question: str, collection) -> str:
108
- context = retrieve_context(question, collection, top_k=1)
109
  return generate_agent_answer(context, question)
110
 
111
  # ─── 5. Gradio Interface Function ─────────────────────────────────────────────
 
105
  return response.text.strip()
106
 
107
  def rag_answer(question: str, collection) -> str:
108
+ context = retrieve_context(question, collection, top_k=2)
109
  return generate_agent_answer(context, question)
110
 
111
  # ─── 5. Gradio Interface Function ─────────────────────────────────────────────