paultltc commited on
Commit
6c0b466
·
1 Parent(s): c502633
Files changed (1) hide show
  1. tool.py +1 -1
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'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]
 
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]