Pimnk commited on
Commit
f771cb8
·
verified ·
1 Parent(s): fe5faaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -352,14 +352,12 @@ def retrieve_chunks(question: str, top_k: int = 20) -> list:
352
  retrieved_nodes = query_engine.retriever.retrieve(question)
353
  log_message(f"Получено {len(retrieved_nodes)} узлов")
354
 
355
- retrieval_params['rerank_threshold'] = rerank_threshold
356
-
357
  reranked_nodes = rerank_nodes(
358
  question,
359
  retrieved_nodes,
360
  reranker,
361
  top_k=top_k,
362
- min_score_threshold=rerank_threshold
363
  )
364
 
365
  chunks_data = []
 
352
  retrieved_nodes = query_engine.retriever.retrieve(question)
353
  log_message(f"Получено {len(retrieved_nodes)} узлов")
354
 
 
 
355
  reranked_nodes = rerank_nodes(
356
  question,
357
  retrieved_nodes,
358
  reranker,
359
  top_k=top_k,
360
+ rerank_threshold=0.5
361
  )
362
 
363
  chunks_data = []