from model.test_model import TestModel
from pathlib import Path
from dataset.dataset import Dataset

if __name__ == "__main__":
    # Loading dataset
    train_dataset = Dataset(Path('data/resized_dataset/train'))
    valid_dataset = Dataset(Path('data/resized_dataset/valid'))
    for i in train_dataset.take(1):
        print(i)

    # Training model
    model = TestModel()
    history = model.fit()
    model.save("./src/model/test_model_final.keras")