Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
| 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 = []
|