diff --git a/docs/index.md b/docs/index.md
index 880f053..9c96f44 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -1,5 +1,5 @@
-#ImaginAIry 🤖ðŸ§
+# ImaginAIry 🤖ðŸ§
**Pythonic AI generation of images and videos**
[![Downloads](https://pepy.tech/badge/imaginairy)](https://pepy.tech/project/imaginairy)
@@ -61,12 +61,12 @@ pip install imaginairy
result.img.save("sun_set_green_valley.png")
```
-
+
---
-## Image Generation Control
+### Image Generation Control
Guide the generation process by providing body poses, depth maps, canny edges, hed boundaries, normal maps, or even QR codes.
#### Body Pose Control
@@ -270,7 +270,7 @@ Replaces existing details in an image. Good to use with --init-image-strength 0.
```
=== "Python"
- ``` py
+ ```py
from imaginairy.api.generate import imagine
from imaginairy.schema import ImaginePrompt, ControlInput, LazyLoadingImage
@@ -323,10 +323,10 @@ allow the tool to generate one for you.
```
=== "Python"
- ``` py
+ ```py
from imaginairy.api.video_sample import generate_video
generate_video(input_path="assets/rocket-wide.png")
```
-
\ No newline at end of file
+
\ No newline at end of file
diff --git a/mkdocs.yml b/mkdocs.yml
index 096be06..383651f 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -35,8 +35,8 @@ plugins:
show_source: true
nav:
- - Home: index.md
- - Docs:
+ - Overview: index.md
+ - API Docs:
- CLI:
- Create Image: docs/CLI/imagine.md
- Create Video: docs/CLI/videogen.md