Update app.py
Browse files
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,
|
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 |
|