Update README.md

README.md

@@ -46,7 +46,7 @@ Thus please make sure to feed only fundus image into the model to obtain reasona
 Use the code below to get started with the model.
 
 ```python
-from transformers import AutoImageProcessor, ResNetForImageClassification
+from transformers import AutoImageProcessor, Swinv2ForImageClassification
 import torch
 from datasets import load_dataset
 
@@ -54,7 +54,7 @@ dataset = load_dataset("huggingface/cats-image")
 image = dataset["test"]["image"][0]
 
 processor = AutoImageProcessor.from_pretrained("pamixsun/swinv2_tiny_for_glaucoma_classification")
-model = ResNetForImageClassification.from_pretrained("pamixsun/swinv2_tiny_for_glaucoma_classification")
+model = Swinv2ForImageClassification.from_pretrained("pamixsun/swinv2_tiny_for_glaucoma_classification")
 
 inputs = processor(image, return_tensors="pt")