Changed model in generative-model:Ex5