ArturG9 commited on
Commit
e15a989
·
verified ·
1 Parent(s): bcfd2df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ os.environ["LANGCHAIN_PROJECT"] = "Lithuanian_Law_RAG_QA"
33
 
34
 
35
 
36
- def create_retriever_from_chroma(vectorstore_path="./docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20):
37
 
38
  model_name = "Alibaba-NLP/gte-base-en-v1.5"
39
  model_kwargs = {'device': 'cpu',
@@ -73,7 +73,7 @@ def create_retriever_from_chroma(vectorstore_path="./docs/chroma/", search_type=
73
  )
74
  st.write("VectorStore is created")
75
 
76
- retriever=vectorstore.as_retriever(search_type = search_type, search_kwargs={"k": k, 'lambda_mult': 0.7})
77
 
78
 
79
 
 
33
 
34
 
35
 
36
+ def create_retriever_from_chroma(vectorstore_path="./docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20,lambda_mult= 0.7):
37
 
38
  model_name = "Alibaba-NLP/gte-base-en-v1.5"
39
  model_kwargs = {'device': 'cpu',
 
73
  )
74
  st.write("VectorStore is created")
75
 
76
+ retriever=vectorstore.as_retriever(search_type = search_type, search_kwargs={"k": k, "lambda_mult": lambda_mult})
77
 
78
 
79