Spaces:
Sleeping
Sleeping
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 |
|