Update README.md

README.md

@@ -34,7 +34,7 @@ model = create_model("resnet18") # pretrained weights loaded from HF
 model = create_model("resnet18", weights="hf://repo_id/filename")
 
 # loading weights from local file
-model = create_model("resnet18", weights="path/to/resnet18/model.npz")
+model = create_model("resnet18", weights="path/to/resnet18/model.safetensors")
 ```
 
 ## ImageNet-1K Results
@@ -42,6 +42,3 @@ model = create_model("resnet18", weights="path/to/resnet18/model.npz")
 Go to https://github.com/riccardomusmeci/mlx-image/blob/main/results/results-imagenet-1k.csv to check every model converted and its performance on ImageNet-1K with different settings.
 
 > **TL;DR** performance is comparable to the original models from PyTorch implementations.
-
-> [!WARNING]
-> More models will be uploaded aligned with MLX improvement by the Apple team.