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