Update app.py
Browse files
app.py
CHANGED
@@ -94,8 +94,8 @@ def main():
|
|
94 |
st.write(css, unsafe_allow_html=True)
|
95 |
|
96 |
|
97 |
-
st.header("Chat with multiple
|
98 |
-
st.markdown("
|
99 |
|
100 |
if "messages" not in st.session_state:
|
101 |
st.session_state["messages"] = [
|
@@ -106,13 +106,13 @@ def main():
|
|
106 |
search_type = st.selectbox(
|
107 |
"Choose search type. Options are [Max marginal relevance search (mmr) , Similarity search (similarity). Default value (mmr)]",
|
108 |
options=["mmr", "similarity"],
|
109 |
-
index=
|
110 |
)
|
111 |
|
112 |
k = st.select_slider(
|
113 |
"Select amount of documents to be retrieved. Default value (5): ",
|
114 |
-
options=list(range(2, 16)),
|
115 |
-
value=
|
116 |
)
|
117 |
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=300, chunk_overlap=30)
|
118 |
|
|
|
94 |
st.write(css, unsafe_allow_html=True)
|
95 |
|
96 |
|
97 |
+
st.header("Chat with multiple Lithuanian Law Documents:")
|
98 |
+
st.markdown("Available Documents: LR_Civil_Code_2022, LR_Constitution_2022, LR_Criminal_Code_2018, LR_Criminal_Procedure_code_2022,LR_Labour_code_2010. P.S it's a shame that there are no newest documents translations... ")
|
99 |
|
100 |
if "messages" not in st.session_state:
|
101 |
st.session_state["messages"] = [
|
|
|
106 |
search_type = st.selectbox(
|
107 |
"Choose search type. Options are [Max marginal relevance search (mmr) , Similarity search (similarity). Default value (mmr)]",
|
108 |
options=["mmr", "similarity"],
|
109 |
+
index=1 # Default to "mmr"
|
110 |
)
|
111 |
|
112 |
k = st.select_slider(
|
113 |
"Select amount of documents to be retrieved. Default value (5): ",
|
114 |
+
options=list(range(2, 16)),
|
115 |
+
value=4
|
116 |
)
|
117 |
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=300, chunk_overlap=30)
|
118 |
|