MNIST Diffusion - ui demo of school project

MNIST Diffusion demo

For a recent school project, I trained a class conditional DDPM diffusion model on the MNIST dataset. This allows for small image generation to look like a specific digit from the dataset. Here you can run the model in your browser using WebGPU acceleration if available. Pick the digit you would like to generate, and the DDIM schedule (or DDPM if you have some patience), and press generate to see your image come to life!