ArturG9 commited on
Commit
6c165fe
1 Parent(s): 9dff375

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -89,7 +89,7 @@ def main():
89
 
90
  st.header("Chat with multiple Lithuanian Law Documents:" ":books:")
91
 
92
- st.markdown("###### Hi, I am Birute (Powered by qwen2-0_5b model), 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.")
93
  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 into English... ")
94
 
95
  if "messages" not in st.session_state:
@@ -164,12 +164,12 @@ def create_conversational_rag_chain(retriever):
164
  model_path = "gemma-2-2b-it-Q8_0.gguf",
165
  seed = 41,
166
  n_gpu_layers=0,
167
- temperature=0.4,
168
  n_ctx=25000,
169
  n_batch=2000,
170
- max_tokens=300,
171
- repeat_penalty=1.9,
172
- last_n_tokens_size = 300,
173
  callback_manager=callback_manager,
174
  verbose=False,
175
  )
 
89
 
90
  st.header("Chat with multiple Lithuanian Law Documents:" ":books:")
91
 
92
+ st.markdown("###### Hi, I am Birute (Powered by gemma-2-2b-it-Q8 model), 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.")
93
  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 into English... ")
94
 
95
  if "messages" not in st.session_state:
 
164
  model_path = "gemma-2-2b-it-Q8_0.gguf",
165
  seed = 41,
166
  n_gpu_layers=0,
167
+ temperature=0.0,
168
  n_ctx=25000,
169
  n_batch=2000,
170
+ max_tokens=250,
171
+ repeat_penalty=1.7,
172
+ last_n_tokens_size = 250,
173
  callback_manager=callback_manager,
174
  verbose=False,
175
  )