Update README.md

README.md

@@ -67,10 +67,13 @@ for t in reversed(range(len(scheduler))):
     sampled_prev_image = prev_image + prev_variance
     image = sampled_prev_image
 
+# process image to PIL
 image_processed = image.cpu().permute(0, 2, 3, 1)
 image_processed = (image_processed + 1.0) * 127.5
 image_processed = image_processed.numpy().astype(np.uint8)
 image_pil = PIL.Image.fromarray(image_processed[0])
+
+# save image
 image_pil.save("test.png")
 ```
 
@@ -81,16 +84,22 @@ Example:
 
 ```python
 from modeling_ddpm import DDPM
+import PIL.Image
+import numpy as np
 
+# load model and scheduler
 ddpm = DDPM.from_pretrained("fusing/ddpm-lsun-bedroom-pipe")
+
+# run pipeline in inference (sample random noise and denoise)
 image = ddpm()
 
-import PIL.Image
-import numpy as np
+# process image to PIL
 image_processed = image.cpu().permute(0, 2, 3, 1)
 image_processed = (image_processed + 1.0) * 127.5
 image_processed = image_processed.numpy().astype(np.uint8)
 image_pil = PIL.Image.fromarray(image_processed[0])
+
+# save image
 image_pil.save("test.png")
 ```