move plot env to top
move plot env to top

chendaniely/multidisciplinary-diffusion-model-experiments,chendaniely/multidisciplinary-diffusion-model-experiments,chendaniely/multidisciplinary-diffusion-model-experiments