nielsr HF staff commited on
Commit
45770ed
1 Parent(s): b5dfd1b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ import torch
26
 
27
  video = list(np.random.randn(8, 3, 224, 224))
28
 
29
- feature_extractor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
30
  model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
31
 
32
  inputs = feature_extractor(video, return_tensors="pt")
 
26
 
27
  video = list(np.random.randn(8, 3, 224, 224))
28
 
29
+ feature_extractor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
30
  model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
31
 
32
  inputs = feature_extractor(video, return_tensors="pt")