Update README.md

README.md

@@ -7,33 +7,49 @@ license_link: https://bria.ai/legal-lobby
 
 # BRIA 3.0 ControlNet Union Model Card
 
+BRIA-3.0 ControlNet-Union, trained on the foundation of [BRIA-4B-Adapt Text-to-Image](https://huggingface.co/briaai/BRIA-4B-Adapt), supports 6 control modes, including depth (0), canny (1), colorgrid (2), recolor (3), tile (4), pose (5). This model can be jointly used with other ControlNets.
 
-BRIA-3.0 ControlNet-Union, trained on the foundation of [BRIA-3.0 Text-to-Image](https://huggingface.co/briaai/BRIA-3.0-TOUCAN)
+This model combines technological innovation with ethical responsibility and legal security, setting a new standard in the AI industry. Bria AI licenses the foundation model with full legal liability coverage. Our dataset does not contain copyrighted materials, such as fictional characters, logos, trademarks, public figures, harmful content, or privacy-infringing content.
 
 [CLICK HERE FOR A DEMO](https://huggingface.co/spaces/briaai/BRIA-2.3-ControlNet-Pose)
 
-[BRIA 3.0](https://huggingface.co/briaai/BRIA-3.0-TOUCAN) was trained from scratch exclusively on licensed data from our esteemed data partners. Therefore, they are safe for commercial use and provide full legal liability coverage for copyright and privacy infringement, as well as harmful content mitigation. That is, our dataset does not contain copyrighted materials, such as fictional characters, logos, trademarks, public figures, harmful content, or privacy-infringing content.
+For more information, please visit our [website](https://bria.ai/).
 
 Join our [Discord community](https://discord.gg/Nxe9YW9zHS) for more information, tutorials, tools, and to connect with other users!
 
 ![controlnet_pose_showoff.png](https://huggingface.co/briaai/BRIA-2.3-ControlNet-Pose/resolve/main/controlnet_pose_showoff.png)
 
+### Get Access
+Interested in BRIA-3.0 ControlNet-Union? Purchase is required to license and access BRIA-3.0 ControlNet-Union, ensuring royalty management with our data partners and full liability coverage for commercial use. 
+
+Are you a startup or a student? We encourage you to apply for our [Startup Program](https://pages.bria.ai/the-visual-generative-ai-platform-for-builders-startups-plan?_gl=1*cqrl81*_ga*MTIxMDI2NzI5OC4xNjk5NTQ3MDAz*_ga_WRN60H46X4*MTcwOTM5OTMzNC4yNzguMC4xNzA5Mzk5MzM0LjYwLjAuMA..) to request access. This program are designed to support emerging businesses and academic pursuits with our cutting-edge technology.
+
+Contact us today to unlock the potential of BRIA-4B-Adapt! By submitting the form above, you agree to BRIA’s [Privacy policy](https://bria.ai/privacy-policy/) and [Terms & conditions](https://bria.ai/terms-and-conditions/).
+
+## Key Features
+
+- **Legally Compliant**: Offers full legal liability coverage for copyright and privacy infringements. Thanks to training on 100% licensed data from leading data partners, we ensure the ethical use of content.
+
+- **Patented Attribution Engine**: Our attribution engine is our way to compensate our data partners, powered by our proprietary and patented algorithms. 
 
+- **Enterprise-Ready**: Specifically designed for business applications, Bria-4B-Adapt delivers high-quality fine-tuning capabilities for generating compliant imagery for a variety of commercial needs.
+
+- **Customizable Technology**: Provides access to source code and weights for extensive customization, catering to specific business requirements.
+
+- **Fully-Automated**: Provides access to fully no-code automatic fine-tuning capabilities on Bria's platform: https://platform.bria.ai/console/tailored-generation.
+
+  
 ### Model Description
 - **Developed by:** BRIA AI
-- **Model type:** [ControlNet](https://huggingface.co/docs/diffusers/using-diffusers/controlnet) for Latent diffusion
-- **License:** [bria-3.0](https://bria.ai/bria-huggingface-model-license-agreement/)
+- **Model type:** Latent Flow-Matching Text-to-Image Model
+- **License:** [Commercial licensing terms & conditions.](https://bria.ai/customer-general-terms-and-conditions)
+- Purchase is required to license and access the model.
 
 - **Model Description:** ControlNet Union for BRIA 3.0  Text-to-Image model. The model generates images guided by text and a conditioned image.
 - **Resources for more information:** [BRIA AI](https://bria.ai/)
 
 
-### Get Access
-BRIA 3.0 ControlNet-Union requires access to BRIA 3.0 Text-to-Image. For more information, [click here](https://huggingface.co/briaai/BRIA-3.0-TOUCAN).
-
-
 ## Control Mode
-
 | Control Mode | Description |
 |:------------:|:-----------:|
 |0|depth
@@ -58,46 +74,46 @@ try:
 except:
     local_dir = '.'
     
-hf_hub_download(repo_id="briaai/BRIA-3.0-TOUCAN", filename='pipeline_bria.py', local_dir=local_dir)
-hf_hub_download(repo_id="briaai/BRIA-3.0-TOUCAN", filename='transformer_bria.py', local_dir=local_dir)
-hf_hub_download(repo_id="briaai/BRIA-3.0-TOUCAN", filename='bria_utils.py', local_dir=local_dir)
+hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='pipeline_bria.py', local_dir=local_dir)
+hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='transformer_bria.py', local_dir=local_dir)
+hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='bria_utils.py', local_dir=local_dir)
 hf_hub_download(repo_id="briaai/BRIA-3.0-ControlNet-Union", filename='pipeline_bria_controlnet.py', local_dir=local_dir)
 hf_hub_download(repo_id="briaai/BRIA-3.0-ControlNet-Union", filename='controlnet_bria.py', local_dir=local_dir)
 
-
 import torch
 from diffusers.utils import load_image
 from controlnet_bria import BriaControlNetModel, BriaMultiControlNetModel
 from pipeline_bria_controlnet import BriaControlNetPipeline
+import PIL.Image as Image
 
-<!-- from diffusers import FluxControlNetPipeline, FluxControlNetModel -->
-
-base_model = 'briaai/BRIA-3.0-TOUCAN'
+base_model = 'briaai/BRIA-4B-Adapt'
 controlnet_model = 'briaai/BRIA-3.0-ControlNet-Union'
 
 controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
-pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)
+pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
 pipe.to("cuda")
 
-control_image_canny = load_image("https://huggingface.co/InstantX/FLUX.1-dev-Controlnet-Union-alpha/resolve/main/images/canny.jpg")
-controlnet_conditioning_scale = 0.5
-control_mode = 0
+control_image_canny = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/canny.jpg")
+controlnet_conditioning_scale = 1.0
+control_mode = 1
 
-width, height = control_image.size
+width, height = control_image_canny.size
 
-prompt = 'A bohemian-style female travel blogger with sun-kissed skin and messy beach waves.'
+prompt = 'In a serene living room, someone rests on a sapphire blue couch, diligently drawing in a rose-tinted notebook, with a sleek black coffee table, a muted green wall, an elegant geometric lamp, and a lush potted palm enhancing the peaceful ambiance.'
 
+generator = torch.Generator(device="cuda").manual_seed(555)
 image = pipe(
     prompt, 
-    control_image=control_image,
+    control_image=control_image_canny,
     control_mode=control_mode,
     width=width,
     height=height,
     controlnet_conditioning_scale=controlnet_conditioning_scale,
-    num_inference_steps=24, 
-    guidance_scale=3.5,
+    num_inference_steps=50, 
+    max_sequence_length=128,
+    guidance_scale=5,
+    generator=generator
 ).images[0]
-image.save("image.jpg")
 ```
 
 # Multi-Controls Inference
@@ -137,12 +153,3 @@ image = pipe(
 ).images[0]
 
 ```
-
-# Resources
-- [InstantX/FLUX.1-dev-Controlnet-Canny](https://huggingface.co/InstantX/FLUX.1-dev-Controlnet-Canny)
-- [InstantX/FLUX.1-dev-Controlnet-Union](https://huggingface.co/InstantX/FLUX.1-dev-Controlnet-Union)
-- [Shakker-Labs/FLUX.1-dev-ControlNet-Depth](https://huggingface.co/Shakker-Labs/FLUX.1-dev-ControlNet-Depth)
-- [Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro](https://huggingface.co/Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro)
-
-# Acknowledgements
-Thanks [zzzzzero](https://github.com/zzzzzero) for help us pointing out some bugs in the training.