Update README.md
Browse files
README.md
CHANGED
|
@@ -127,7 +127,7 @@ print(query_embedding.shape, document_embeddings.shape)
|
|
| 127 |
# Compute cosine similarity between the query and documents
|
| 128 |
similarity = model.similarity(query_embedding, document_embeddings)
|
| 129 |
print(similarity)
|
| 130 |
-
# tensor([[0.
|
| 131 |
```
|
| 132 |
|
| 133 |
### With Transformers
|
|
|
|
| 127 |
# Compute cosine similarity between the query and documents
|
| 128 |
similarity = model.similarity(query_embedding, document_embeddings)
|
| 129 |
print(similarity)
|
| 130 |
+
# tensor([[0.6968, 0.7818, 0.7165, 0.8374]])
|
| 131 |
```
|
| 132 |
|
| 133 |
### With Transformers
|