From 45770ed49890b989166f5f87fd121cb3b70171cb Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Tue, 6 Dec 2022 09:56:23 +0000 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index e5df683..67f398b 100644 --- a/README.md +++ b/README.md @@ -26,7 +26,7 @@ import torch video = list(np.random.randn(8, 3, 224, 224)) -feature_extractor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") +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")