MCG-NJU
/

videomae-small-finetuned-ssv2

Video Classification

Inference Endpoints

Model card Files Files and versions Community

sandstorm12 commited on Apr 25, 2023

Commit

b2159d0

•

1 Parent(s): 79e8333

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))
-feature_extractor = VideoMAEFeatureExtractor.from_pretrained("sandstorm12/videomae-small-finetuned-ssv2")
-model = VideoMAEForVideoClassification.from_pretrained("sandstorm12/videomae-small-finetuned-ssv2")
 inputs = feature_extractor(video, return_tensors="pt")

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