update README.md

README.md

@@ -22,32 +22,3 @@ model-index:
 
 Trained on CIFAR-10 (10 classes). Weights saved as a plain PyTorch `state_dict` (`pytorch_model.bin`).
 Architecture is defined in `vit_model.py` (uses `timm`).
-
-## Usage
-
-```python
-import torch, json
-from huggingface_hub import hf_hub_download
-import importlib.util
-
-repo_id = "roylvzn/vit-cifar10"
-
-# fetch files
-weights_path = hf_hub_download(repo_id, "pytorch_model.bin")
-model_py     = hf_hub_download(repo_id, "vit_model.py")
-classes_path = hf_hub_download(repo_id, "classes.json")
-
-# import vit_model.py dynamically
-spec = importlib.util.spec_from_file_location("vit_model", model_py)
-vm = importlib.util.module_from_spec(spec); spec.loader.exec_module(vm)
-
-# build model and load weights
-model = vm.ViTModel(num_classes=10, pretrained=False)
-state = torch.load(weights_path, map_location="cpu")
-model.load_state_dict(state)
-model.eval()
-
-with open(classes_path) as f:
-    classes = json.load(f)
-
-# inference expects 224x224 ImageNet-normalized tensors