Fix usage example

README.md

@@ -7655,7 +7655,9 @@ NOTE: A good uniform scalar quantization range to use with this model (and which
 You can use the sentence-transformers package to use any of the snowflake-arctic-embed models. Here's an example for `snowflake-arctic-embed-m-v1.5`.
 
 ```python
+import torch
 from sentence_transformers import SentenceTransformer
+from torch.nn.functional import normalize
 
 # Model constant.
 MODEL_ID = "Snowflake/snowflake-arctic-embed-m-v1.5"
@@ -7696,8 +7698,8 @@ for query, query_scores in zip(queries, scores):
 #
 
 #### Variation: Truncated Embeddings ####
-query_embeddings_256 = normalize(query_embeddings[:, :256])
-doument_embeddings_256 = normalize(doument_embeddings[:, :256])
+query_embeddings_256 = normalize(torch.from_numpy(query_embeddings)[:, :256])
+doument_embeddings_256 = normalize(torch.from_numpy(document_embeddings)[:, :256])
 scores_256 = query_embeddings_256 @ doument_embeddings_256.T
 
 # Pretty-print the results.