ArturG9 commited on
Commit
2c751e4
1 Parent(s): 82e97af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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 PDFs :books:")
98
- st.markdown("Hi, I am Qwen, chat mmodel, based on respublic of Lithuania law document. Write you question and press enter to start chat.")
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=0 # 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)), # Creates a list [2, 3, 4, ..., 15]
115
- value=5 # Default 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