Update README.md

README.md

@@ -34,7 +34,7 @@ transform = ImageNetTransform(train=False, img_size=224)
 x = transform(read_rgb("cat.png"))
 x = mx.expand_dims(x, 0)
 
-model = create_model("resnet18")
+model = create_model("wide_resnet50_2")
 model.eval()
 
 logits = model(x)
@@ -51,13 +51,13 @@ x = transform(read_rgb("cat.png"))
 x = mx.expand_dims(x, 0)
 
 # first option
-model = create_model("resnet18", num_classes=0)
+model = create_model("wide_resnet50_2", num_classes=0)
 model.eval()
 
 embeds = model(x)
 
 # second option
-model = create_model("resnet18")
+model = create_model("wide_resnet50_2")
 model.eval()
 
 embeds = model.features(x)