Geralt-Targaryen commited on
Commit
8883c48
·
verified ·
1 Parent(s): 376fa52

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -78,8 +78,8 @@ documents = [
78
  ]
79
 
80
  # Encode the query and documents
81
- query_embedding = model.encode_query(query)
82
- document_embeddings = model.encode_document(documents)
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