Ketansomewhere
/

FER_2013_Conditional_Diffusion

Conditional Diffusion

Model card Files Files and versions

Ketansomewhere commited on Feb 24

Commit

6f947a9

·

verified ·

1 Parent(s): 876610b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -145,7 +145,7 @@ def create_image_grid(images, grid_size, save_path):
     return grid_img
 if __name__ == "__main__":
-    repo_id = "Ketansomewhere/King"
     num_classes = 7  # Adjust to your number of classes
     batch_size = 64
     num_inference_steps = 1000  # Can be as low as 50 for faster generation

     return grid_img
 if __name__ == "__main__":
+    repo_id = "Ketansomewhere/FER_2013_Conditional_Diffusion"
     num_classes = 7  # Adjust to your number of classes
     batch_size = 64
     num_inference_steps = 1000  # Can be as low as 50 for faster generation