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)