Update README.md

README.md

@@ -9,7 +9,7 @@ datasets:
 - imagenet-1k
 library_name: mlx-image
 ---
-#  vit_large_patch16_224.swag_lin-mlxim
+#  vit_large_patch16_224.swag_lin
 
 A [Vision Transformer](https://arxiv.org/abs/2010.11929v2) image classification model. Weights are learned with [SWAG](https://arxiv.org/abs/2201.08371) on ImageNet-1k data.
 
@@ -32,7 +32,7 @@ transform = ImageNetTransform(train=False, img_size=224)
 x = transform(read_rgb("cat.png"))
 x = mx.expand_dims(x, 0)
 
-model = create_model(" vit_large_patch16_224.swag_lin-mlxim")
+model = create_model("vit_large_patch16_224.swag_lin")
 model.eval()
 
 logits = model(x)
@@ -49,16 +49,16 @@ x = transform(read_rgb("cat.png"))
 x = mx.expand_dims(x, 0)
 
 # first option
-model = create_model(" vit_large_patch16_224.swag_lin-mlxim", num_classes=0)
+model = create_model("vit_large_patch16_224.swag_lin", num_classes=0)
 model.eval()
 
 embeds = model(x)
 
 # second option
-model = create_model(" vit_large_patch16_224.swag_lin-mlxim")
+model = create_model("vit_large_patch16_224.swag_lin")
 model.eval()
 
-embeds = model.features(x)
+embeds = model.get_features(x)
 ```