this is a test model trained with 10 epochs