Update README.md

README.md

@@ -1,4 +1,15 @@
-lightweight GAN trained on glid-3 orbs for demo I'm working on.
+---
+language: en
+tags:
+- lightweightgan
+license: apache-2.0
+datasets:
+- johnowhitaker/glid3_orbs
+---
+
+# orbgan
+
+lightweight GAN trained on my glid-3 orbs for demo I'm working on.
 
 Training notebook: https://colab.research.google.com/drive/16o1TdrxnQ54Msbr813XfPVsnEt2QTRAa?usp=sharing
 
@@ -6,9 +17,9 @@ Inference notebook: https://colab.research.google.com/drive/1e7dR2dptM8F1xhRcyy-
 
 Please credit if you use this, and feedback on the code is welcomed :)
 
-Demo:
+## Demo:
 
-from lightweight_gan import Generator
+```from lightweight_gan import Generator
 import torch
 from matplotlib import pyplot as plt
 from huggingface_hub import PyTorchModelHubMixin
@@ -31,4 +42,4 @@ ims = gan_new(torch.randn(n_rows**2, 256)).clamp_(0., 1.)
 fig, axs = plt.subplots(n_rows, n_rows, figsize=(9, 9))
 for i, ax in enumerate(axs.flatten()):
   ax.imshow(ims[i].permute(1, 2, 0).detach().cpu().numpy())
-plt.tight_layout()
+plt.tight_layout()```