Update README.md

README.md

@@ -7,6 +7,7 @@ tags:
 - diffusion
 - Conditional Diffusion
 ---
+## Diffusion model trained on FER 2013 dataset. 
 
 
 Here is Custom Pipeline for Class conditioned diffusion model. For training script, pipeline, tutorial nb and sampling please check my Github Repo:- https://github.com/KetanMann/Class_Conditioned_Diffusion_Training_Script