fcakyon commited on
Commit
3753aee
1 Parent(s): d460be8

fix a typo in code snippet

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -20,13 +20,13 @@ You can use the raw model for video classification into one of the 174 possible
20
  Here is how to use this model to classify a video:
21
 
22
  ```python
23
- from transformers import TimesformerFeatureExtractor, TimesformerForVideoClassification
24
  import numpy as np
25
  import torch
26
 
27
  video = list(np.random.randn(16, 3, 448, 448))
28
 
29
- feature_extractor = TimesformerFeatureExtractor.from_pretrained("facebook/timesformer-hr-finetuned-ssv2")
30
  model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-hr-finetuned-ssv22")
31
 
32
  inputs = feature_extractor(video, return_tensors="pt")
 
20
  Here is how to use this model to classify a video:
21
 
22
  ```python
23
+ from transformers import VideoMAEFeatureExtractor, TimesformerForVideoClassification
24
  import numpy as np
25
  import torch
26
 
27
  video = list(np.random.randn(16, 3, 448, 448))
28
 
29
+ feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
30
  model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-hr-finetuned-ssv22")
31
 
32
  inputs = feature_extractor(video, return_tensors="pt")