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 MaxvitTinyWeights
+from lucid.models.weights import MaxViTTinyWeights
 
 # default tag
 model = models.maxvit_tiny_cls(pretrained=True)
 
 # explicit tag (enum or string)
-model = models.maxvit_tiny_cls(weights=MaxvitTinyWeights.IN1K)
+model = models.maxvit_tiny_cls(weights=MaxViTTinyWeights.IN1K)
 model = models.maxvit_tiny_cls(pretrained="IN1K")
 
 # preprocessing travels with the weights
-weights = MaxvitTinyWeights.IN1K
+weights = MaxViTTinyWeights.IN1K
 preprocess = weights.transforms()
 logits = model(preprocess(image)[None]).logits
 ```