Update app.py

app.py

@@ -17,7 +17,6 @@ folder = snapshot_download(repo_id="umaiku/faiss_index", repo_type="dataset", lo
 embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-small")
 
 vector_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
-retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": 0.75})
 
 def respond(
     message,
@@ -26,9 +25,11 @@ def respond(
     max_tokens,
     temperature,
     top_p,
+    score,
 ):
     messages = [{"role": "system", "content": system_message}]
 
+    retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": score})
     document = retriever.invoke(message)
 
     if document == []:
@@ -77,6 +78,7 @@ demo = gr.ChatInterface(
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
+        gr.Slider(minimum=0, maximum=1, value=0.7, step=0.1, label="Score Threshold"),
     ],
     description="# 📜 ALexI: Artificial Legal Intelligence for Swiss Jurisprudence",
 )