Lhumpal commited on
Commit
4533586
·
verified ·
1 Parent(s): a7e6ea2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,7 +41,7 @@ def retrieve(query, vectorstore, top_k=5):
41
  docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
42
 
43
  # Filter results based on score threshold
44
- filtered_docs_and_scores = [(doc.page_content, float(score)) for doc, score in docs_and_scores if float(score) <= 0.75]
45
 
46
  # Separate docs from the (doc, score) tuples
47
  docs_content = [doc for doc, _ in filtered_docs_and_scores]
@@ -133,7 +133,7 @@ async def chat(request: ChatRequest):
133
  config=GenerateContentConfig(
134
  system_instruction=[request.system_message],
135
  max_output_tokens=150,
136
- temperature=0.9
137
  ),
138
  )
139
 
 
41
  docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
42
 
43
  # Filter results based on score threshold
44
+ filtered_docs_and_scores = [(doc.page_content, float(score)) for doc, score in docs_and_scores if float(score) <= 0.8]
45
 
46
  # Separate docs from the (doc, score) tuples
47
  docs_content = [doc for doc, _ in filtered_docs_and_scores]
 
133
  config=GenerateContentConfig(
134
  system_instruction=[request.system_message],
135
  max_output_tokens=150,
136
+ temperature=1.5
137
  ),
138
  )
139