Update README.md

README.md

@@ -136,7 +136,7 @@ from pipeline_bria_controlnet import BriaControlNetPipeline
 import PIL.Image as Image
 
 base_model = 'briaai/BRIA-3.1'
-controlnet_model = 'briaai/BRIA-3.0-ControlNet-Union'
+controlnet_model = 'briaai/BRIA-3.1-ControlNet-Union'
 
 controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 controlnet = BriaMultiControlNetModel([controlnet])
@@ -144,8 +144,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/images/colorgrid.jpg")
-control_image_pose = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/images/pose.jpg")
+control_image_colorgrid = load_image("https://huggingface.co/briaai/BRIA-3.1-ControlNet-Union/resolve/main/images/colorgrid.jpg")
+control_image_pose = load_image("https://huggingface.co/briaai/BRIA-3.1-ControlNet-Union/resolve/main/images/pose.jpg")
 
 control_image = [control_image_colorgrid, control_image_pose]
 controlnet_conditioning_scale = [0.5, 0.5]