Update README.md

README.md

@@ -1,13 +1,24 @@
 ---
 tags:
 - text-to-image
+- stable-diffusion
 - lora
 - diffusers
-- template:diffusion-lora
+- image-generation
+- flux
+- safetensors
 widget:
-- output:
-    url: images/image.png
-  text: change to illustration style
+- text: >-
+    change to illustration style
+  output:
+    url: images/example1.png
+- text: >-
+    change to illustration style
+  output:
+    url: images/example2.png
+- text: change to illustration style
+  output:
+    url: images/example3.png
 base_model: black-forest-labs/FLUX.1-Kontext-dev
 instance_prompt: illustration style
 license: other
@@ -17,15 +28,39 @@ license_link: >-
 ---
 # FLUX.1-Kontext-dev-LoRA-Illustration-Style
 
-<Gallery />
+This is an illustration style style LoRA trained on FLUX.1-Kontext-dev by [木木木亘](https://www.liblib.art/userpage/8d5ade1f022c453089e473c7e39dbb35/publish).
 
+## Showcases
+<Gallery />
 
 ## Trigger words
 
 You should use `illustration style` to trigger the image generation.
 
 
-## Download model
+## Inference
+
+```python
+import torch
+from diffusers import FluxKontextPipeline
+from diffusers.utils import load_image
+
+pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("Shakker-Labs/FLUX.1-Kontext-dev-LoRA-Illustration-Style", weight_name="FLUX-kontext-lora-illustration-style.safetensors")
+pipe.fuse_lora(lora_scale=1.0)
+pipe.to("cuda")
+
+input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
+prompt = "change to illustration style"
 
+image = pipe(
+    image=input_image,
+    prompt=prompt, 
+    num_inference_steps=24, 
+    guidance_scale=2.5,
+).images[0]
+image.save(f"example.png")
+```
 
-[Download](/Shakker-Labs/FLUX.1-Kontext-dev-LoRA-Illustration-Style/tree/main) them in the Files & versions tab.
+## Acknowledgements
+This model is trained by our copyrighted users [木木木亘](https://www.liblib.art/userpage/8d5ade1f022c453089e473c7e39dbb35/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).