Fix usage example: import weights enum from lucid.models.weights

README.md

@@ -35,17 +35,17 @@ converted to Lucid-native safetensors.
 
 ```python
 import lucid.models as models
-from lucid.models.vision.resnet import MaxvitSmallWeights
+from lucid.models.weights import MaxViTSmallWeights
 
 # default tag
 model = models.maxvit_small_cls(pretrained=True)
 
 # explicit tag (enum or string)
-model = models.maxvit_small_cls(weights=MaxvitSmallWeights.IN1K)
+model = models.maxvit_small_cls(weights=MaxViTSmallWeights.IN1K)
 model = models.maxvit_small_cls(pretrained="IN1K")
 
 # preprocessing travels with the weights
-weights = MaxvitSmallWeights.IN1K
+weights = MaxViTSmallWeights.IN1K
 preprocess = weights.transforms()
 logits = model(preprocess(image)[None]).logits
 ```