Add model file and modify training scripts