use smaller README image (#131)

* add smaller version of README image

* use smaller image in README

* required to display inline on (Test)PyPI

* add some modules requiring docstrings to docs

README.md

@@ -1,6 +1,6 @@
 # DNA Diffusion
 
-<img src='https://github.com/pinellolab/DNA-Diffusion/raw/1413ec9c974d54a1e2885a250540dc89955d10c0/docs/images/diff_first.gif'> </img>
+<img src='https://raw.githubusercontent.com/pinellolab/DNA-Diffusion/main/docs/images/diff_first_lossy.gif?inline=true'> </img>
 
 <p align="center">
     <em>Generative modeling of regulatory DNA sequences with diffusion probabilistic models.</em>

docs/reference/dnadiffusion.md

@@ -1,6 +1,47 @@
 # Reference
 
-::: dnadiffusion
+## dnadiffusion
+
+::: dnadiffusion.trainer
+    options:
+      show_root_heading: true
+      show_source: false
+
+::: dnadiffusion.configs
+    options:
+      show_root_heading: true
+      show_source: false
+
+::: dnadiffusion.sample
+    options:
+      show_root_heading: true
+      show_source: false
+
+## data
+
+::: dnadiffusion.data.dataloader
+    options:
+      show_root_heading: true
+      show_source: false
+
+## models
+
+::: dnadiffusion.models.modules
+    options:
+      show_root_heading: true
+      show_source: false
+
+::: dnadiffusion.models.training_modules
+    options:
+      show_root_heading: true
+      show_source: false
+
+::: dnadiffusion.models.networks
+    options:
+      show_root_heading: true
+      show_source: false
+
+::: dnadiffusion.models.unet
     options:
-      show_root_heading: false
+      show_root_heading: true
       show_source: false