ajusting the threshold score

Ramyrahmeni · Jun 15, 2024 · 5cc1f68 · 5cc1f68
1 parent 9b13b3e
commit 5cc1f68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app.py b/app.py
@@ -147,7 +147,7 @@ def retrieve_relevant_resources(query: str,
     dot_scores = util.dot_score(query_embedding, embeddings)[0]
 
     # Get the number of scores above a threshold (e.g., 0.5)
-    valid_scores = dot_scores > 0.1
+    valid_scores = dot_scores 
     cnt = valid_scores.sum().item()
 
     # Get top K scores and their indices