Merge pull request #873 from roboflow/fix/confidence-not-applied-in-c…

…lassification-models Fix confidence not applied when inferring classification models
roboflow · Dec 12, 2024 · a24d655 · a24d655
2 parents 25aa233 + 3c9f5ac
commit a24d655
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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(