Update README.md

README.md

@@ -38,7 +38,7 @@ model.eval()
 logits = model(x)
 ```
 
-You can also use the embeds from last conv layer:
+You can also use the embeds from layer before head:
 ```python
 from mlxim.model import create_model
 from mlxim.io import read_rgb