Update README.md

README.md

@@ -44,9 +44,15 @@ BRIA-4B-Adapt-ControlNet-Union requires access to BRIA-4B-Adapt Text-to-Image. F
 
 
 ```python
-pip install diffusers==0.30.2, hf_hub_download
+
 ```
 
+### Installations
+
+```bash
+pip install -qr https://huggingface.co/briaai/BRIA-4B-Adapt/resolve/main/requirements.txt
+pip install diffusers==0.30.2, hf_hub_download
+```
 
 ```python
 from huggingface_hub import hf_hub_download
@@ -61,7 +67,10 @@ hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='transformer_bria.py',
 hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='bria_utils.py', local_dir=local_dir)
 hf_hub_download(repo_id="briaai/BRIA-4B-Adapt-ControlNet-Union", filename='pipeline_bria_controlnet.py', local_dir=local_dir)
 hf_hub_download(repo_id="briaai/BRIA-4B-Adapt-ControlNet-Union", filename='controlnet_bria.py', local_dir=local_dir)
+```
 
+## Inference
+```python
 import torch
 from diffusers.utils import load_image
 from controlnet_bria import BriaControlNetModel
@@ -124,21 +133,6 @@ print(image)
 
 # Multi-Controls Inference
 ```python
-from huggingface_hub import hf_hub_download
-import os
-
-try:
-    local_dir = os.path.dirname(__file__)
-except:
-    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
@@ -154,8 +148,8 @@ controlnet = BriaMultiControlNetModel([controlnet])
 pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
 pipe.to("cuda")
 
-control_image_colorgrid = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/colorgrid.jpg")
-control_image_pose = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/pose.jpg")
+control_image_colorgrid = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/images/colorgrid.jpg")
+control_image_pose = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/images/pose.jpg")
 
 control_image = [control_image_colorgrid, control_image_pose]
 controlnet_conditioning_scale = [0.5, 0.5]