facebook
/

timesformer-base-finetuned-k400

Video Classification

Inference Endpoints

Model card Files Files and versions Community

Update README.md

#4

by tiansz - opened May 4, 2023

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ from transformers import AutoImageProcessor, TimesformerForVideoClassification
 import numpy as np
 import torch
-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")

 import numpy as np
 import torch
+video = list(np.random.randn(8, 3, 224, 224).astype(np.uint8))
 processor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
 model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")