# Codeblock 20
model = UNet().to(DEVICE)
noise_scheduler = NoiseScheduler()