diff --git a/notebooks/utils.py b/notebooks/utils.py
index c7ee8ef..4268dd5 100644
--- a/notebooks/utils.py
+++ b/notebooks/utils.py
@@ -653,7 +653,7 @@ class names to use, by default None
 
     # Define classifier
     custom_downstream_model = CatBoostClassifier(
-        iterations=8000,
+        iterations=2000,  # Not too high to avoid too large model size
         depth=8,
         early_stopping_rounds=50,
         loss_function=loss_function,