MCG-NJU
/

videomae-small-finetuned-ssv2

Video Classification

Inference Endpoints

Model card Files Files and versions Community

Update README.md

#1

by sandstorm12 - opened Apr 25, 2023

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

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")