fix syntax error

train_main.py

@@ -16,7 +16,8 @@ def process_dataset_with_augmentation(df, is_training=False):
     # 1. Calculate Statistics for Balancing (Only needed for training)
     if is_training:
         class_counts = df['target'].value_counts().to_dict()
-        max_count = max(class_counts.values())        print(f"  [Augmentation] Balancing classes to match majority count: {max_count}")
+        max_count = max(class_counts.values())        
+        print(f"  [Augmentation] Balancing classes to match majority count: {max_count}")
 
     total = len(df)