Update README.md

README.md

@@ -20,7 +20,29 @@ This model was trained on a dataset of 272 high-quality images of aurora.
 #### How to use
 
 ```python
-# You can include sample code which will be formatted
+# Clone this model 
+git clone https://huggingface.co/huggan/fastgan-few-shot-aurora/
+
+def load_generator(model_name_or_path):
+    generator = Generator(in_channels=256, out_channels=3)
+    generator = generator.from_pretrained(model_name_or_path, in_channels=256, out_channels=3)
+    _ = generator.eval()
+
+    return generator
+    
+def _denormalize(input: torch.Tensor) -> torch.Tensor:
+    return (input * 127.5) + 127.5
+    
+# Load generator
+generator = load_generator("fastgan-few-shot-aurora")
+
+# Generate a random noise image
+noise = torch.zeros(1, 256, 1, 1, device=device).normal_(0.0, 1.0)
+with torch.no_grad():
+  gan_images, _ = generator(noise)
+  
+gan_images = _denormalize(gan_images.detach())
+save_image(gan_images, "sample.png", nrow=1, normalize=True)
 ```
 
 #### Limitations and bias