Spaces:
Runtime error
Runtime error
Commit
·
116f44b
1
Parent(s):
262cb62
Fix
Browse files- src/SemanticSearch.py +1 -1
src/SemanticSearch.py
CHANGED
|
@@ -204,7 +204,7 @@ class SemanticSearch:
|
|
| 204 |
|
| 205 |
if rerank:
|
| 206 |
|
| 207 |
-
documents = [f"'{item.lower()}' ({cls[
|
| 208 |
|
| 209 |
rank_results = self.get_reranked_results(nlq, documents)
|
| 210 |
|
|
|
|
| 204 |
|
| 205 |
if rerank:
|
| 206 |
|
| 207 |
+
documents = [f"'{item.lower()}' ({cls['class_label']})" for item, cls in zip(results["documents"][0], results["metadatas"][0])]
|
| 208 |
|
| 209 |
rank_results = self.get_reranked_results(nlq, documents)
|
| 210 |
|