docs: fix the code snippets

README.md

@@ -238,26 +238,32 @@ model = SentenceTransformer("jinaai/jina-embeddings-v4", trust_remote_code=True)
 # 1. Retrieval Task
 # ========================
 # Encode query
-query_embedding = model.encode(
+query_embeddings = model.encode(
     sentences=["Overview of climate change impacts on coastal cities"],
     task="retrieval",
     prompt_name="query",
-)[0]
+)
+
+print(f"query_embeddings.shape = {query_embeddings.shape}")
 
 # Encode passage (text)
-passage_embedding = model.encode(
+passage_embeddings = model.encode(
     sentences=[
         "Climate change has led to rising sea levels, increased frequency of extreme weather events..."
     ],
     task="retrieval",
     prompt_name="passage",
-)[0]
+)
+
+print(f"passage_embeddings.shape = {passage_embeddings.shape}")
 
 # Encode image/document
-image_embedding = model.encode(
+image_embeddings = model.encode(
     sentences=["https://i.ibb.co/nQNGqL0/beach1.jpg"],
     task="retrieval",
-)[0]
+)
+
+print(f"image_embeddings.shape = {image_embeddings.shape}")
 
 # ========================
 # 2. Text Matching Task
@@ -281,7 +287,7 @@ text_embeddings = model.encode(sentences=texts, task="text-matching")
 # ========================
 
 # Encode query
-query_embedding = model.encode(
+query_embeddings = model.encode(
     sentences=["Find a function that prints a greeting message to the console"],
     task="code",
     prompt_name="query",
@@ -293,6 +299,25 @@ code_embeddings = model.encode(
     task="code",
     prompt_name="passage",
 )
+
+# ========================
+# 4. Use multivectors
+# ========================
+
+multivector_text_embeddings = model.encode(
+    sentences=texts,
+    task="retrieval",
+    prompt_name="query",
+    return_multivector=True,
+)
+
+images = ["https://i.ibb.co/nQNGqL0/beach1.jpg", "https://i.ibb.co/r5w8hG8/beach2.jpg"]
+
+multivector_image_embeddings = model.encode(
+    sentences=images,
+    task="retrieval",
+    return_multivector=True,
+)
 ```
 </details>