facebook
/

timesformer-base-finetuned-k400

Video Classification

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Jan 2, 2023

Commit

8aaf40e

•

1 Parent(s): e040485

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -26,10 +26,10 @@ import torch
 video = list(np.random.randn(8, 3, 224, 224))
-feature_extractor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
 model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
-inputs = feature_extractor(video, return_tensors="pt")
 with torch.no_grad():
   outputs = model(**inputs)

 video = list(np.random.randn(8, 3, 224, 224))
+processor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
 model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
+inputs = processor(video, return_tensors="pt")
 with torch.no_grad():
   outputs = model(**inputs)