Added the model traniner

Author: Ananddd06

app.py

@@ -3,13 +3,15 @@
 from networksecurity.components.data_ingestion import DataIngestion
 from networksecurity.components.data_validation import DataValidation
 from networksecurity.components.data_tranformation import DataTransformation
+from networksecurity.components.model_trainer import ModelTrainer
 from networksecurity.logger.customlogger import Custom_Logger
 from networksecurity.exception.exception import CustomException
 from networksecurity.entity.config_entity import (
     DataIngestionConfigEntity,
     DataValidationConfigEntity,
     DataTransformationConfigEntity,
     TrainingPipelineConfigEntity,
+    ModelTrainerConfigEntity,
 )
 from networksecurity.entity.artifact_entity import DataIngestionArtifact , DataValidationArtifact
 
@@ -47,6 +49,19 @@
         data_transformation_artifact = data_transformation.initiate_data_transformation()
         logger.info("✅ Data transformation completed successfully.")
         logger.info(f"📊 Data TransformationArtifact: {data_transformation_artifact}")
+
+        # Model Training
+        model_trainer_config = ModelTrainerConfigEntity(
+            training_pipeline_config=training_pipeline_config
+        )
+        model_trainer = ModelTrainer(
+            model_trainer_config=model_trainer_config,
+            data_transformation_config=data_transformation_config,
+            data_transformation_artifact=data_transformation_artifact,
+        )
+        model_trainer_artifact = model_trainer.initiate_model_trainer()
+        logger.info("✅ Model training completed successfully.")
+        logger.info(f"🤖 Model Trainer Artifact: {model_trainer_artifact}")
     except Exception as e:
         logger.error(f"❌ Unexpected error in main pipeline: {e}")
         raise CustomException(e, sys)