Update README.md

README.md

@@ -28,16 +28,16 @@ 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, VideoMAEForVideoClassification
+from transformers import VideoMAEImageProcessor, VideoMAEForVideoClassification
 import numpy as np
 import torch
 
 video = list(np.random.randn(16, 3, 224, 224))
 
-feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-ssv2")
+processor = VideoMAEImageProcessor.from_pretrained("MCG-NJU/videomae-base-finetuned-ssv2")
 model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base-finetuned-ssv2")
 
-inputs = feature_extractor(video, return_tensors="pt")
+inputs = processor(video, return_tensors="pt")
 
 with torch.no_grad():
   outputs = model(**inputs)