Update README.md

README.md

@@ -4,7 +4,9 @@ tags:
 - stable-diffusion
 - lora
 - diffusers
-- template:sd-lora
+- image-generation
+- flux
+- safetensors
 widget:
 - text: >-
     sketched style,A stick-figure-style robot, and some sci-fi machines, with
@@ -25,18 +27,46 @@ license: other
 license_name: flux-1-dev-non-commercial-license
 license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
 ---
-# FLUX.1-dev-LoRA-Children's-Simple-Sketch
+# FLUX.1-dev-LoRA-Children-Simple-Sketch
 
-<Gallery />
+This is a LoRA trained on FLUX.1-dev for Children's Simple-Sketch by [Hongke](https://www.shakker.ai/userpage/b1f1d4128cec474c8085576e6c9fb761/publish) on [Shakker AI](https://www.shakker.ai/modelinfo/eb54f3880dd04cc4b9002d31753a12ec?from=personal_page).
+
+<div class="container">
+  <img src="https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-live-3D/resolve/main/poster.jpg" width="1024"/>
+</div>
 
 
+## Showcases
+<Gallery />
+
 ## Trigger words
+You should use `sketched style` to trigger the image generation. The recommended scale is `0.8` to `1.5` in diffusers.
+
+
+## Inference
+
+```python
+import torch
+from diffusers import FluxPipeline
+
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
+pipe.fuse_lora(lora_scale=1.5)
+pipe.to("cuda")
+
+prompt = "sketched style,A joyful girl with balloons floats above a city wearing a hat and striped pants,"
 
-You should use `sketched style` to trigger the image generation.
+image = pipe(prompt, 
+             num_inference_steps=24, 
+             guidance_scale=3.5,
+            ).images[0]
+image.save(f"example.png")
+```
 
+## Online Inference
 
-## Download model
+You can also run this model at [Shakker AI](https://www.shakker.ai/modelinfo/eb54f3880dd04cc4b9002d31753a12ec?from=personal_page), where we provide an online interface to generate images.
 
-Weights for this model are available in Safetensors format.
 
-[Download](/Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch/tree/main) them in the Files & versions tab.
+## Acknowledgements
+This model is trained by our copyrighted users [Hongke](https://www.shakker.ai/userpage/b1f1d4128cec474c8085576e6c9fb761/publish). We release this model under permissions. The model follows [flux-1-dev-non-commercial-license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md).