Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ def main():
|
|
104 |
|
105 |
st.markdown("Hi, I am Birute, chat assistant, based on republic of Lithuania law documents. You can choose below information retrieval type and how many documents you want to be retrieved.")
|
106 |
search_type = st.selectbox("Choose search type. Options are [Max marginal relevance search(mmr) , Similarity search (similarity). Default value (mmr)]", options =["mmr", "similarity"], index ="mmr")
|
107 |
-
k = st.select_slider("Select amount of documents to be retrieved. Default value(5) : ", options=[2, 15],
|
108 |
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=450, chunk_overlap=20)
|
109 |
|
110 |
if user_question := st.text_input("Ask a question about your documents:"):
|
|
|
104 |
|
105 |
st.markdown("Hi, I am Birute, chat assistant, based on republic of Lithuania law documents. You can choose below information retrieval type and how many documents you want to be retrieved.")
|
106 |
search_type = st.selectbox("Choose search type. Options are [Max marginal relevance search(mmr) , Similarity search (similarity). Default value (mmr)]", options =["mmr", "similarity"], index ="mmr")
|
107 |
+
k = st.select_slider("Select amount of documents to be retrieved. Default value(5) : ", options=[2, 15], value = 5)
|
108 |
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=450, chunk_overlap=20)
|
109 |
|
110 |
if user_question := st.text_input("Ask a question about your documents:"):
|