umaiku commited on
Commit
47761f1
·
verified ·
1 Parent(s): c3d6f33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -43,7 +43,8 @@ def respond(
43
  print(system_message)
44
 
45
  # retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": score, "k": 10})
46
- retriever = vector_db.as_retriever(search_type="mmr")
 
47
  documents = retriever.invoke(message)
48
 
49
  spacer = " \n"
@@ -113,7 +114,7 @@ demo = gr.ChatInterface(
113
  additional_inputs=[
114
  gr.Textbox(value="You are an assistant in Swiss Jurisprudence cases.", label="System message"),
115
  gr.Slider(minimum=1, maximum=24000, value=5000, step=1, label="Max new tokens"),
116
- gr.Slider(minimum=0.1, maximum=4.0, value=0.1, step=0.1, label="Temperature"),
117
  gr.Slider(
118
  minimum=0.1,
119
  maximum=1.0,
 
43
  print(system_message)
44
 
45
  # retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": score, "k": 10})
46
+ retriever = vector_db.as_retriever(search_type="similarity_score", search_kwargs={"k": 10})
47
+ # retriever = vector_db.as_retriever(search_type="mmr")
48
  documents = retriever.invoke(message)
49
 
50
  spacer = " \n"
 
114
  additional_inputs=[
115
  gr.Textbox(value="You are an assistant in Swiss Jurisprudence cases.", label="System message"),
116
  gr.Slider(minimum=1, maximum=24000, value=5000, step=1, label="Max new tokens"),
117
+ gr.Slider(minimum=0.1, maximum=4.0, value=0, step=0.1, label="Temperature"),
118
  gr.Slider(
119
  minimum=0.1,
120
  maximum=1.0,