Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -34,8 +34,8 @@ import torch
34
 
35
  video = list(np.random.randn(16, 3, 224, 224))
36
 
37
- feature_extractor = VideoMAEFeatureExtractor.from_pretrained("sandstorm12/videomae-small-finetuned-ssv2")
38
- model = VideoMAEForVideoClassification.from_pretrained("sandstorm12/videomae-small-finetuned-ssv2")
39
 
40
  inputs = feature_extractor(video, return_tensors="pt")
41
 
 
34
 
35
  video = list(np.random.randn(16, 3, 224, 224))
36
 
37
+ feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-small-finetuned-ssv2")
38
+ model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-small-finetuned-ssv2")
39
 
40
  inputs = feature_extractor(video, return_tensors="pt")
41