Update README.md

README.md

@@ -2,22 +2,20 @@
 license: other
 ---
 
-# ControlNet Canny for Flux
+![ControlNet Canny for Flux](https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/light/controlnet-canny-header-rev1.png?raw=true)
 
 This repository provides a checkpoint with trained ControlNet Canny model for
 [FLUX.1-dev model](https://huggingface.co/black-forest-labs/FLUX.1-dev) by Black Forest Labs
 
 # Training details
 [XLabs AI](https://github.com/XLabs-AI) team is happy to publish fune-tuning Flux scripts, including:
-
 - **LoRA** 🔥
 - **ControlNet** 🔥
 
 [See our github](https://github.com/XLabs-AI/x-flux) for train script and train configs.
 
-# Training Dataset
+# Training dataset
 Dataset has the following format for the training process:
-
 ```
 ├── images/
 │    ├── 1.png
@@ -28,7 +26,38 @@ Dataset has the following format for the training process:
 ```
 A .json file contains "caption" field with a text prompt.
 
-# License
+# Inference
+
+To test our checkpoints, use commands presented below.
+
+### LoRA
+
+```bash
+python3 demo_lora_inference.py \
+    --checkpoint lora.safetensors \
+    --prompt "A handsome girl in a suit covered with bold tattoos and holding a pistol. fantasy style, natural photo cinematic"
+```
 
+![Example Picture 0](https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/examples/picture-1-rev1.png?raw=true)
+
+### ControlNet (Canny)
+```bash
+python3 demo_controlnet_inference.py \
+    --checkpoint controlnet.safetensors \
+    --control_image "input_image.jpg" \
+    --prompt "a handsome viking man with white hair, cinematic, MM full HD"
+```
+![Example Picture 1](https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/examples/picture-3-rev1.png?raw=true)
+```bash
+python3 demo_controlnet_inference.py \
+    --checkpoint controlnet.safetensors \
+    --control_image "input_image.jpg" \
+    --prompt "a dark evil mysterius house with ghosts, cinematic, MM full HD"
+```
+
+![Example Picture 2](https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/examples/picture-2-rev1.png?raw=true)
+
+
+# License
 controlnet.safetensors falls under the [FLUX.1 [dev]](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md) Non-Commercial License<br/>
 controlnet.bin falls under the [FLUX.1 [dev]](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md) Non-Commercial License