Update README.md

README.md

@@ -39,17 +39,17 @@ fine-tuned versions on a task that interests you.
 Here is how to use this model:
 
 ```python
->>> from transformers import AutoFeatureExtractor, ResNetForImageClassification
+>>> from transformers import AutoImageProcessor, AutoModelForImageClassification
 >>> import torch
 >>> from datasets import load_dataset
 
 >>> 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")
+>>> image_processor = AutoImageProcessor.from_pretrained("microsoft/resnet-18")
+>>> model = AutoModelForImageClassification.from_pretrained("microsoft/resnet-18")
 
->>> inputs = feature_extractor(image, return_tensors="pt")
+>>> inputs = image_processor(image, return_tensors="pt")
 
 >>> with torch.no_grad():
 ...     logits = model(**inputs).logits