Update README.md

README.md

@@ -38,7 +38,7 @@ batch = tokenizer.batch_encode_plus(
                 truncation=True
             )
 
-# Compute the embeddings
+# Compute the embeddings and keep only the _**[CLS]**_ embedding (the first token)
 
 outputs = model(**batch, output_hidden_states=True, return_dict=True)