fix default env variables numbers

model.py

@@ -35,11 +35,11 @@ class BertClassifier(LabelStudioMLBase):
     def __init__(self, project_id=None, label_config=None, **kwargs):
         super(BertClassifier, self).__init__(project_id=project_id, label_config=label_config)
         
-        # Load training configuration from environment variables
-        self.learning_rate = float(os.getenv('LEARNING_RATE'))
-        self.num_train_epochs = int(os.getenv('NUM_TRAIN_EPOCHS'))
-        self.weight_decay = float(os.getenv('WEIGHT_DECAY'))
-        self.start_training_threshold = int(os.getenv('START_TRAINING_EACH_N_UPDATES'))
+        # Load training configuration from environment variables with defaults
+        self.learning_rate = float(os.getenv('LEARNING_RATE', '2e-5'))
+        self.num_train_epochs = int(os.getenv('NUM_TRAIN_EPOCHS', '20'))
+        self.weight_decay = float(os.getenv('WEIGHT_DECAY', '0.01'))
+        self.start_training_threshold = int(os.getenv('START_TRAINING_EACH_N_UPDATES', '1'))
         
         logger.info("=== Training Configuration ===")
         logger.info(f"✓ Learning rate: {self.learning_rate}")