The data set if correctly organised. Might need to experiment with lower batch size or change the model.