Update README.md
Browse files
README.md
CHANGED
|
@@ -78,8 +78,8 @@ documents = [
|
|
| 78 |
]
|
| 79 |
|
| 80 |
# Encode the query and documents
|
| 81 |
-
query_embedding = model.
|
| 82 |
-
document_embeddings = model.
|
| 83 |
print(query_embedding.shape, document_embeddings.shape)
|
| 84 |
# (4096,) (4, 4096)
|
| 85 |
|
|
|
|
| 78 |
]
|
| 79 |
|
| 80 |
# Encode the query and documents
|
| 81 |
+
query_embedding = model.encode(query)
|
| 82 |
+
document_embeddings = model.encode(documents)
|
| 83 |
print(query_embedding.shape, document_embeddings.shape)
|
| 84 |
# (4096,) (4, 4096)
|
| 85 |
|