Update README.md

README.md

@@ -160,4 +160,6 @@ if __name__ == "__main__":
         plt.title(f'Class {class_label}')
         plt.savefig(os.path.join(save_dir, f"grid_image_class{class_label}.png"))
         plt.show()
-```
+```
+
+Also, check this nb for the above implementation *testing.py* .