ArturG9 commited on
Commit
1f3521c
·
verified ·
1 Parent(s): de7666e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -103,8 +103,8 @@ def main():
103
  ]
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.radio("Choose search type. Options are [Max marginal relevance search(mmr) , Similarity search (similarity). Default value (mmr)]", options=["mmr", "similarity"], value="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:"):
 
103
  ]
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.radio("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], index = 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:"):