diff --git a/README.md b/README.md index 217c548..1cdbeee 100644 --- a/README.md +++ b/README.md @@ -34,8 +34,8 @@ import torch video = list(np.random.randn(16, 3, 224, 224)) -feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base") -model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base") +feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") +model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") inputs = feature_extractor(video, return_tensors="pt")