MCG-NJU
/

videomae-small-finetuned-kinetics

Video Classification

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Apr 22, 2023

Commit

1bf2b82

•

1 Parent(s): 8fc0b88

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -34,8 +34,8 @@ import torch
 video = list(np.random.randn(16, 3, 224, 224))
-processor = VideoMAEImageProcessor.from_pretrained("sandstorm12/videomae-small-finetuned-kinetics")
-model = VideoMAEForVideoClassification.from_pretrained("sandstorm12/videomae-small-finetuned-kinetics")
 inputs = processor(video, return_tensors="pt")

 video = list(np.random.randn(16, 3, 224, 224))
+processor = VideoMAEImageProcessor.from_pretrained("MCG-NJU/videomae-small-finetuned-kinetics")
+model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-small-finetuned-kinetics")
 inputs = processor(video, return_tensors="pt")