Update README.md

README.md

@@ -4,7 +4,9 @@ tags:
 - stable-diffusion
 - lora
 - diffusers
-- template:sd-lora
+- image-generation
+- flux
+- safetensors
 widget:
 - text: logo,Minimalist,Make a picture of a camera with the word 'Me'
   output:
@@ -23,20 +25,42 @@ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICE
 ---
 # FLUX.1-dev-LoRA-Logo-Design
 
+This is a LoRA (Logo-Design) trained on FLUX.1-dev by [CJim](https://www.shakker.ai/userpage/b43e5cc7fcd64d2bbde42c6e889267cc/publish) on [Shakker AI](https://www.shakker.ai/modelinfo/0355942f9bd140a99e371ba5731894e8?from=personal_page).
+<div class="container">
+  <img src="./poster.jpg" width="1024"/>
+</div>
+
+## Showcases
 <Gallery />
 
 
 ## Trigger words
 
-You should use `wablogo` to trigger the image generation.
+You should use `wablogo, logo, Minimalist` as trigger words. The recommended scale is `0.8` in diffusers.
+
+## Inference
+
+```python
+import torch
+from diffusers import FluxPipeline
 
-You should use `logo` to trigger the image generation.
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design", weight_name="FLUX-dev-lora-Logo-Design.safetensors")
+pipe.fuse_lora(lora_scale=0.8)
+pipe.to("cuda")
 
-You should use `Minimalist` to trigger the image generation.
+prompt = "logo,Minimalist,A bunch of grapes and a wine glass"
+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 download this model at [Shakker AI](https://www.shakker.ai/modelinfo/0355942f9bd140a99e371ba5731894e8?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-Logo-Design/tree/main) them in the Files & versions tab.
+## Acknowledgements
+This model is trained by our copyrighted users [CJim](https://www.shakker.ai/userpage/b43e5cc7fcd64d2bbde42c6e889267cc/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).