From 8aaf40ea7d3d282dcb0a5dea01a198320d15d6c0 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Mon, 2 Jan 2023 11:43:07 +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 67f398b..92b72b9 100644 --- a/README.md +++ b/README.md @@ -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") +processor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400") model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400") -inputs = feature_extractor(video, return_tensors="pt") +inputs = processor(video, return_tensors="pt") with torch.no_grad(): outputs = model(**inputs)