The trained model used for generation.
Achieved 72% training accuracy and 63% validation accuracy.