diff --git a/README.md b/README.md
index e34be95..e5df683 100644
--- a/README.md
+++ b/README.md
@@ -20,13 +20,13 @@ You can use the raw model for video classification into one of the 400 possible
 Here is how to use this model to classify a video:
 
 ```python
-from transformers import VideoMAEFeatureExtractor, TimesformerForVideoClassification
+from transformers import AutoImageProcessor, TimesformerForVideoClassification
 import numpy as np
 import torch
 
 video = list(np.random.randn(8, 3, 224, 224))
 
-feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
+feature_extractor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
 model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
 
 inputs = feature_extractor(video, return_tensors="pt")