khang119966 commited on
Commit
0d06e91
1 Parent(s): 59c925a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -124,11 +124,10 @@ scores = processor.score_multi_vector(query_embeddings, image_embeddings)
124
  max_scores, max_indices = torch.max(scores, dim=1)
125
  # In ra kết quả cho mỗi câu hỏi
126
  for i, query in enumerate(queries):
127
- image_name = images[max_indices[i]]
128
  print(f"Câu hỏi: '{query}'")
129
  print(f"Điểm số: {max_scores[i].item()}\n")
130
  plt.figure(figsize=(5,5))
131
- plt.imshow(image_name)
132
  plt.show()
133
  ```
134
 
 
124
  max_scores, max_indices = torch.max(scores, dim=1)
125
  # In ra kết quả cho mỗi câu hỏi
126
  for i, query in enumerate(queries):
 
127
  print(f"Câu hỏi: '{query}'")
128
  print(f"Điểm số: {max_scores[i].item()}\n")
129
  plt.figure(figsize=(5,5))
130
+ plt.imshow(images[max_indices[i]])
131
  plt.show()
132
  ```
133