diff --git a/preprocessor_config.json b/preprocessor_config.json
index 70fbc14..34a1af4 100644
--- a/preprocessor_config.json
+++ b/preprocessor_config.json
@@ -1,6 +1,7 @@
 {
   "do_normalize": true,
   "do_resize": true,
+  "feature_extractor_type": "ViTFeatureExtractor",
   "image_mean": [
     0.5,
     0.5,