Spaces:
Runtime error
Runtime error
File size: 1,731 Bytes
a0f1e38 b9a1fab 821ffc1 cf8627d a0f1e38 b9a1fab 821ffc1 cf8627d a0f1e38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
from src.imagecolorization.pipeline.stage01_data_ingestion import DataIngestionPipeline
from src.imagecolorization.pipeline.stage02_data_transformation import DataTransformationPipeline
from src.imagecolorization.pipeline.stage_03_model_building import ModelBuildingPipeline
from src.imagecolorization.pipeline.stage_04_model_trainer import ModelTrainerPipeline
from src.imagecolorization.logging import logger
STAGE_NAME = 'Data Ingestion Config'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
data_ingestion = DataIngestionPipeline()
data_ingestion.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
STAGE_NAME = 'Data Tranasformation Config'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
data_transformation = DataTransformationPipeline()
data_transformation.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
STAGE_NAME = 'Model Building Config'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
model_building = ModelBuildingPipeline()
model_building.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
STAGE_NAME = 'Model Training Config'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
model_trianer = ModelTrainerPipeline()
model_trianer.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e |