diff --git a/inference/core/models/classification_base.py b/inference/core/models/classification_base.py index ac56089e65..6422c4d87b 100644 --- a/inference/core/models/classification_base.py +++ b/inference/core/models/classification_base.py @@ -162,6 +162,7 @@ def infer( disable_preproc_grayscale=disable_preproc_grayscale, disable_preproc_static_crop=disable_preproc_static_crop, return_image_dims=return_image_dims, + **kwargs, ) def postprocess(