Update README.md
Browse files
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(
|
| 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 |
|