new model run with 150 epochs and learning rate of 0.0003. Model checkpointing, early stopping and reduce learning rate on plateau callbacks have been added to the model training component. Model evaluation component has been updated to log validation loss and accuracy to mlflow and save it in scores.json file.