Update README.md

README.md

@@ -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("swin_tiny_patch4_window7_224/")
+model = create_model("swin_tiny_patch4_window7_224")
 model.eval()
 
 logits = model(x)
@@ -49,13 +49,13 @@ x = transform(read_rgb("cat.png"))
 x = mx.expand_dims(x, 0)
 
 # first option
-model = create_model("swin_tiny_patch4_window7_224/", num_classes=0)
+model = create_model("swin_tiny_patch4_window7_224", num_classes=0)
 model.eval()
 
 embeds = model(x)
 
 # second option
-model = create_model("swin_tiny_patch4_window7_224/")
+model = create_model("swin_tiny_patch4_window7_224")
 model.eval()
 
 embeds = model.get_features(x)