daniel-dona commited on
Commit
116f44b
·
1 Parent(s): 262cb62
Files changed (1) hide show
  1. 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["class_label"]})" for item, cls in zip(results["documents"][0], results["metadatas"][0])]
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