Update README.md

README.md

@@ -46,17 +46,20 @@ Then you can use the model like this:
 ```python
 from sentence_transformers import SentenceTransformer
 
-# Each input text should start with "query: " or "passage: ".
-# For tasks other than retrieval, you can simply use the "query: " prefix.
-input_texts = [
-  "query: Когда был спущен на воду первый миноносец «Спокойный»?",
-  "query: Есть ли нефть в Удмуртии?",
-  "passage: Спокойный (эсминец)\nЗачислен в списки ВМФ СССР 19 августа 1952 года.",
-  "passage: Нефтепоисковые работы в Удмуртии были начаты сразу после Второй мировой войны в 1945 году и продолжаются по сей день. Добыча нефти началась в 1967 году."
+queries = [
+  "Когда был спущен на воду первый миноносец «Спокойный»?",
+  "Есть ли нефть в Удмуртии?"
+]
+passages = [
+  "Спокойный (эсминец)\nЗачислен в списки ВМФ СССР 19 августа 1952 года.",
+  "Нефтепоисковые работы в Удмуртии были начаты сразу после Второй мировой войны в 1945 году и продолжаются по сей день. Добыча нефти началась в 1967 году."
 ]
 
 model = SentenceTransformer("deepvk/USER-base")
-embeddings = model.encode(input_texts, normalize_embeddings=True)
+# Prompt should be specified according to the task (either 'query' or 'passage').
+passage_embeddings = model.encode(passages, normalize_embeddings=True, prompt_name='passage')
+# For tasks other than retrieval, you can simply use the `query` prompt, which is set by default.
+query_embeddings = model.encode(queries, normalize_embeddings=True)
 ```
 
 However, you can use model directly with [`transformers`](https://huggingface.co/docs/transformers/en/index)
@@ -75,7 +78,7 @@ def average_pool(
     )
     return last_hidden.sum(dim=1) / attention_mask.sum(dim=1)[..., None]
 
-# Each input text should start with "query: " or "passage: ".
+# You should manually add prompts when using the model directly. Each input text should start with "query: " or "passage: ".
 # For tasks other than retrieval, you can simply use the "query: " prefix.
 input_texts = [
   "query: Когда был спущен на воду первый миноносец «Спокойный»?",