nielsr HF staff commited on
Commit
0695923
1 Parent(s): cb29842

Fix code example

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -46,8 +46,8 @@ Here is how to use this model:
46
  >>> dataset = load_dataset("huggingface/cats-image")
47
  >>> image = dataset["test"]["image"][0]
48
 
49
- >>> feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
50
- >>> model = ResNetForImageClassification.from_pretrained("microsoft/resnet-50")
51
 
52
  >>> inputs = feature_extractor(image, return_tensors="pt")
53
 
 
46
  >>> dataset = load_dataset("huggingface/cats-image")
47
  >>> image = dataset["test"]["image"][0]
48
 
49
+ >>> feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-18")
50
+ >>> model = ResNetForImageClassification.from_pretrained("microsoft/resnet-18")
51
 
52
  >>> inputs = feature_extractor(image, return_tensors="pt")
53