From b5dfd1b5182b96dff247be1565bd5521625dfd54 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Tue, 6 Dec 2022 09:56:11 +0000 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index e34be95..e5df683 100644 --- a/README.md +++ b/README.md @@ -20,13 +20,13 @@ You can use the raw model for video classification into one of the 400 possible Here is how to use this model to classify a video: ```python -from transformers import VideoMAEFeatureExtractor, TimesformerForVideoClassification +from transformers import AutoImageProcessor, TimesformerForVideoClassification import numpy as np import torch video = list(np.random.randn(8, 3, 224, 224)) -feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") +feature_extractor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400") inputs = feature_extractor(video, return_tensors="pt")