microsoft
/

resnet-18

Image Classification

Model card Files Files and versions

nielsr HF Staff commited on Jun 7, 2022

Commit

0695923

·

1 Parent(s): cb29842

Fix code example

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,8 +46,8 @@ Here is how to use this model:
 >>> dataset = load_dataset("huggingface/cats-image")
 >>> image = dataset["test"]["image"][0]
->>> feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
->>> model = ResNetForImageClassification.from_pretrained("microsoft/resnet-50")
 >>> inputs = feature_extractor(image, return_tensors="pt")

 >>> dataset = load_dataset("huggingface/cats-image")
 >>> image = dataset["test"]["image"][0]
+>>> feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-18")
+>>> model = ResNetForImageClassification.from_pretrained("microsoft/resnet-18")
 >>> inputs = feature_extractor(image, return_tensors="pt")