chainyo commited on
Commit
48351af
1 Parent(s): d842208

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -1
README.md CHANGED
@@ -1,5 +1,27 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
3
  ---
4
 
5
- DocuGAN model
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ library_name: pytorch
4
+ tags:
5
+ - gan
6
+ - sngan
7
+ - huggan
8
+ dataset: ChainYo/rvl-cdip-invoice
9
  ---
10
 
11
+ ## Model description
12
+
13
+ SN-GAN implementation with PyTorch-Lightning to generate Documents.
14
+
15
+ ## Generated samples
16
+
17
+ <img src="https://github.com/ChainYo/docugan/blob/master/documents_samples.png" width="400" height="1200">
18
+
19
+ Project repository: [DocuGAN](https://github.com/ChainYo/docugan).
20
+
21
+ ## Usage
22
+
23
+ You see the genration tool on HuggingFace by trying the [space demo](https://huggingface.co/spaces/ChainYo/DocuGAN).
24
+
25
+ ## Training data
26
+
27
+ For training, I used the invoices subpart of `RVL-CDIP` dataset. Find the full dataset [here](https://huggingface.co/datasets/ChainYo/rvl-cdip)