Darka001 commited on
Commit
35f3267
·
verified ·
1 Parent(s): ec13266

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -97,9 +97,9 @@ db3 = Chroma(persist_directory="chroma/", embedding_function=instructor_embeddin
97
 
98
 
99
 
100
- retriever = db3.as_retriever(search_type="mmr",
101
  search_kwargs={
102
- "k": 20})
103
 
104
  #retriever = db3.as_retriever(search_kwargs={"k":15})
105
  # Get pre-written rag prompt
@@ -224,7 +224,7 @@ def generate_answer(message, history, checkbox, subject_input, rubric, date_beg,
224
 
225
  result['answer'] =result['answer'].split("[/INST]")[-1].strip()
226
  formatted_results = format_result(result)
227
- return result
228
 
229
 
230
  def change_group(check_value):
 
97
 
98
 
99
 
100
+ retriever = db3.as_retriever(
101
  search_kwargs={
102
+ "k": 15})
103
 
104
  #retriever = db3.as_retriever(search_kwargs={"k":15})
105
  # Get pre-written rag prompt
 
224
 
225
  result['answer'] =result['answer'].split("[/INST]")[-1].strip()
226
  formatted_results = format_result(result)
227
+ return formatted_results
228
 
229
 
230
  def change_group(check_value):