Extracting the class label of generated images from multiclass image data using DDPM

Hi,

 I'm currently performing unconditional image generation. My input is multiclass image data with 10 class of images, expected output is generated images for each class. I'm able to generate images, but not able to extract/connect the class label into which the generated image belong to.

Thanks and regards