readme: ---
  license: mit
  library_name: pytorch
  tags:
  - gan
  - sngan
  - huggan
  - unconditional-image-generation
  datasets: 
  - ChainYo/rvl-cdip-invoice
  ---
  
  ## Model description
  
  SN-GAN implementation with PyTorch-Lightning to generate Documents.
  
  ## Generated samples
  
  <img src="https://raw.githubusercontent.com/ChainYo/docugan/master/documents_samples.png" width="400" height="1200">
  
  Project repository: [DocuGAN](https://github.com/ChainYo/docugan).
  
  ## Usage
  
  You can see the tool to generate document on HuggingFace by trying the [space demo](https://huggingface.co/spaces/ChainYo/DocuGAN).
  
  ## Training data
  
  For training, I used the invoices subpart of `RVL-CDIP` dataset. Find the full dataset [here](https://huggingface.co/datasets/ChainYo/rvl-cdip)
  