pamixsun
/

swinv2_tiny_for_glaucoma_classification

Image Classification

Model card Files Files and versions

pamixsun commited on Aug 8, 2023

Commit

4f38308

·

1 Parent(s): a6ccf31

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -53,8 +53,8 @@ from datasets import load_dataset
 dataset = load_dataset("huggingface/cats-image")
 image = dataset["test"]["image"][0]
-processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
-model = ResNetForImageClassification.from_pretrained("microsoft/resnet-50")
 inputs = processor(image, return_tensors="pt")

 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")
 inputs = processor(image, return_tensors="pt")