Update README code snippet

README.md

@@ -38,15 +38,30 @@ Since we're a small company, this model is only released under a non-commercial
 ```python
 from sentence_transformers import SentenceTransformer
 
-model = SentenceTransformer("zeroentropy/zembed-1", trust_remote_code=True)
-
-sentences = [
-    "What is backpropagation?",
-    "Backpropagation computes gradients of the loss with respect to each weight.",
+# Initialize model
+model = SentenceTransformer(
+    "zeroentropy/zembed-1",
+    trust_remote_code=True,
+    # truncate_dim=640,  # Optional: Reduce dimensionality from 2560 to {1280, 640, 320, 160, 80, 40} 
+    model_kwargs={"torch_dtype": "bfloat16"},
+)
+
+# Define query and documents
+query = "What is backpropagation?"
+documents = [
+    "Backpropagation is a fundamental algorithm for training neural networks by computing gradients.",
+    "Gradient descent is used to optimize model parameters during the training process.",
+    "Neural network training relies on efficient computation of derivatives through backpropagation.",
 ]
 
-embeddings = model.encode(sentences)
-print(embeddings.shape)
+# Encode query and documents (uses task-specific prompts automatically)
+query_embeddings = model.encode_query(query)
+document_embeddings = model.encode_document(documents)
+# (2560,) (3, 2560)
+
+# Compute cosine similarities
+similarities = model.similarity(query_embeddings, document_embeddings)
+# tensor([[0.7525, 0.5670, 0.6835]])
 ```
 
 The model can also be used through ZeroEntropy's [/models/embed](https://docs.zeroentropy.dev/api-reference/models/embed) endpoint.

config_sentence_transformers.json

@@ -5,6 +5,6 @@
     "document": "<|im_start|>system\ndocument<|im_end|>\n<|im_start|>user\n"
   },
   "suffix": "<|im_end|>\n",
-  "default_prompt_name": null,
+  "default_prompt_name": "document",
   "similarity_fn_name": "cosine"
 }