diff --git a/image_prediction/redai_adapter/model.py b/image_prediction/redai_adapter/model.py
index a0ac3a8..05efc3a 100644
--- a/image_prediction/redai_adapter/model.py
+++ b/image_prediction/redai_adapter/model.py
@@ -12,7 +12,7 @@ class PredictionModelHandle:
 
     def predict(self, *args, **kwargs):
         tensor, valid_mask = self.__prep_images(*args, **kwargs)
-        predictions = self.__predict(tensor, **kwargs)
+        predictions = self.__predict(tensor)
         return [p if v else None for p, v in zip(predictions, valid_mask)]
 
     def __call__(self, *args, **kwargs):