teddyllm commited on
Commit
c783ba9
1 Parent(s): efb12ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ conv_chain = (
64
  def chat_gen(message, history, return_buffer=True):
65
  buffer = ""
66
 
67
- doc_retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": 0.2, "k": 10})
68
  retrieved_docs = doc_retriever.invoke(message)
69
  print(len(retrieved_docs))
70
  print(retrieved_docs)
 
64
  def chat_gen(message, history, return_buffer=True):
65
  buffer = ""
66
 
67
+ doc_retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": 0.2, "k": 5})
68
  retrieved_docs = doc_retriever.invoke(message)
69
  print(len(retrieved_docs))
70
  print(retrieved_docs)