joortif commited on
Commit
2bc9ce3
·
verified ·
1 Parent(s): 946d2c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def test_rag_reranking(query, ranker):
52
 
53
  # ✅ Corregido: pasar solo lista de strings
54
  context_strings = [str(c) for c in context]
55
- reranked = ranker.rank(query, context_strings, 1)
56
  print(f"🏅 Resultado del reranker: {reranked}")
57
 
58
  # ✅ Seguridad en el acceso al mejor contexto
 
52
 
53
  # ✅ Corregido: pasar solo lista de strings
54
  context_strings = [str(c) for c in context]
55
+ reranked = ranker.rank(query=query, documents=context_strings, top_k=1)
56
  print(f"🏅 Resultado del reranker: {reranked}")
57
 
58
  # ✅ Seguridad en el acceso al mejor contexto