Update README.md

README.md

@@ -40,7 +40,9 @@ batch = tokenizer.batch_encode_plus(
 
 # Compute the embeddings and keep only the _**[CLS]**_ embedding (the first token)
 
-outputs = model(**batch, output_hidden_states=True, return_dict=True)
+# Get raw embeddings (no gradients)
+with torch.no_grad():
+    outputs = model(**batch, output_hidden_states=True, return_dict=True)
 
 embeddings = outputs.last_hidden_state[:,0]