'''
This file is used only for evaluation without the need to train the model
i.e Loads the best model and evaluates on your test dataset
'''

from gcg.components import load_data, build_model, evaluate_model
from gcg.utils import load_from_checkpoint, logging
from gcg import config

logging.info("Initiated evaluation pipeline")

logging.info(f"Loading data from {config.data_path}")
X_train, X_test, y_train, y_test = load_data(config.data_path, config.image_size)

logging.info("Building model...")
model = build_model(input_shape=config.image_size, num_classes=7)

logging.info("Loading the model checkpoint...")
model = load_from_checkpoint(model, config.model_path)

logging.info("Evaluating the model on test set...")
evaluate_model(model, X_test, y_test)