Update README.md

README.md

@@ -40,7 +40,7 @@ model.eval()
 logits = model(x)
 ```
 
-You can also use the embeds from last conv layer:
+You can also use the embeds from the last conv layer:
 ```python
 from mlxim.model import create_model
 from mlxim.io import read_rgb
@@ -60,7 +60,7 @@ embeds = model(x)
 model = create_model("resnet18")
 model.eval()
 
-embeds = model.features(x)
+embeds = model.get_features(x)
 ```
 
 ## Model Comparison