Spaces:
Running
Running
fix typo
Browse files
tool.py
CHANGED
|
@@ -285,7 +285,7 @@ class VisualRAGTool(Tool):
|
|
| 285 |
top_k_idx = scores.topk(k).indices.tolist()
|
| 286 |
|
| 287 |
print("Top Scores:")
|
| 288 |
-
[print(f
|
| 289 |
|
| 290 |
# Get the top k results
|
| 291 |
results = [self.pages[idx] for idx in top_k_idx]
|
|
|
|
| 285 |
top_k_idx = scores.topk(k).indices.tolist()
|
| 286 |
|
| 287 |
print("Top Scores:")
|
| 288 |
+
[print(f"Page {self.pages[idx].metadata.get('page_id')}: {scores[idx]}") for idx in top_k_idx]
|
| 289 |
|
| 290 |
# Get the top k results
|
| 291 |
results = [self.pages[idx] for idx in top_k_idx]
|