HUANG-Stephanie commited on
Commit
fe7b387
1 Parent(s): cd99189

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -153,7 +153,7 @@ async def search_by_cv(file: UploadFile = File(...), k: int = 10):
153
  scores = retriever_evaluator.evaluate(qs, ds)
154
 
155
  # Trouver les indices des résultats les plus pertinents
156
- top_k_indices = scores.argsort(axis=1)[0][-k-1:0][::-1]
157
 
158
  # Préparer les résultats sous forme d'images
159
  results = []
 
153
  scores = retriever_evaluator.evaluate(qs, ds)
154
 
155
  # Trouver les indices des résultats les plus pertinents
156
+ top_k_indices = scores.argsort(axis=1)[0][-k-1:-1][::-1]
157
 
158
  # Préparer les résultats sous forme d'images
159
  results = []