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]
|