chinese_lantern_riddles_v2_sft

Sleeping

ytyeung commited on Feb 27

Commit

9f764a9

•

1 Parent(s): 8c968bb

Update threshold

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,7 +119,7 @@ def answer(input_text,context=None):
 # helper function for RAG
 def helper_rag(text):
-    docs_out = vectordb.similarity_search_with_relevance_scores(text,k=1)
     #docs_out = vectordb.max_marginal_relevance_search(text,k=5,fetch_k = 20, lambda_mult = 0.5)
     context = []
     for doc in docs_out:

 # helper function for RAG
 def helper_rag(text):
+    docs_out = vectordb.similarity_search_with_relevance_scores(text,k=3)
     #docs_out = vectordb.max_marginal_relevance_search(text,k=5,fetch_k = 20, lambda_mult = 0.5)
     context = []
     for doc in docs_out: