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