Update README.md

README.md

@@ -60,7 +60,7 @@ embedding_mean = torch.mean(last_hidden_state[0], dim=0)
 print(embedding_mean.shape) # expect to be 768
 
 # embedding with max pooling
-embedding_max = torch.max(last_hidden_states[0], dim=0)[0]
+embedding_max = torch.max(last_hidden_state[0], dim=0)[0]
 print(embedding_max.shape) # expect to be 768
 ```