Update README.md

README.md

@@ -30,7 +30,7 @@ Note: if you are using this for downstream inference, adjust the num_classes bas
 import torch
 import torch.nn as nn
 from huggingface_hub import PyTorchModelHubMixin
-from visionfm_hf import VisionTransformer
+from vit import vit_large_patch16
 
 class RetFiner(nn.Module, PyTorchModelHubMixin):
     """RetFiner: Fine-tuned ViT models for retinal image analysis"""