Update README.md

README.md

@@ -14,4 +14,8 @@ A demo space for https://huggingface.co/johnowhitaker/orbgan_e1
 
 Runs a lightwieght-gan trained on https://huggingface.co/datasets/johnowhitaker/glid3_orbs. 
 See model card (https://huggingface.co/johnowhitaker/orbgan_e1) for training and inference scripts.
-n_rows controls how many it makes at a time, the result being an n_rows by n_rows grid. Keep it small please :)
+n_rows controls how many it makes at a time, the result being an n_rows by n_rows grid.
+
+I've also added two similar models (orbgan_dark and orbgan_light) - the only difference is that they were trained on subsets of the training data to get more consistent outputs.
+
+Feel free to reach out if you have questions or want to replicate this for yourself :)