Photoroom interview question

Some basic diffusion model related coding challenge.