Update README.md

README.md

@@ -76,6 +76,8 @@ passages = ["The capital of France is Paris.", "China's capital is Beijing."]
 inputs = tokenizer(query, return_tensors="pt")
 query_embeddings = model(**inputs, return_dict=True)[0][:, 0, :dim]
 
+print(query_embeddings.shape) # torch.Size([1, 32])
+
 passage_inputs = tokenizer(passages, return_tensors="pt", padding=True, truncation=True)
 passage_embeddings = model(**passage_inputs, return_dict=True)[0][:, 0, :dim]