improving vector search

Ramyrahmeni · Jun 19, 2024 · 7782749 · 7782749
1 parent 40a72ce
commit 7782749
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app.py b/app.py
@@ -108,7 +108,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 > 0.4
     cnt = valid_scores.sum().item()
 
     # Get top K scores and their indices