Spaces:
Running
Running
Update search_utils.py
Browse files- search_utils.py +1 -3
search_utils.py
CHANGED
@@ -168,9 +168,7 @@ class SemanticSearch:
|
|
168 |
results['similarity'] = 1 - (distances / 2)
|
169 |
|
170 |
# Deduplicate and sort results
|
171 |
-
results = results.drop_duplicates(subset=["title", "source"])
|
172 |
-
.sort_values("similarity", ascending=False)
|
173 |
-
.head(top_k)
|
174 |
|
175 |
return results.reset_index(drop=True)
|
176 |
|
|
|
168 |
results['similarity'] = 1 - (distances / 2)
|
169 |
|
170 |
# Deduplicate and sort results
|
171 |
+
results = results.drop_duplicates(subset=["title", "source"]).sort_values("similarity", ascending=False).head(top_k)
|
|
|
|
|
172 |
|
173 |
return results.reset_index(drop=True)
|
174 |
|