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

README.md

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