Update README.md

README.md

@@ -46,12 +46,13 @@ 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, Swinv2ForImageClassification
+import cv2
 import torch
-from datasets import load_dataset
 
-dataset = load_dataset("huggingface/cats-image")
-image = dataset["test"]["image"][0]
+from transformers import AutoImageProcessor, Swinv2ForImageClassification
+
+image = cv2.imread('./example.jpg')
+image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
 processor = AutoImageProcessor.from_pretrained("pamixsun/swinv2_tiny_for_glaucoma_classification")
 model = Swinv2ForImageClassification.from_pretrained("pamixsun/swinv2_tiny_for_glaucoma_classification")