TalHach61 commited on
Commit
3e884f7
·
verified ·
1 Parent(s): 55a678f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -75,7 +75,7 @@ from pipeline_bria_controlnet import BriaControlNetPipeline
75
  import PIL.Image as Image
76
 
77
  base_model = 'briaai/BRIA-4B-Adapt'
78
- controlnet_model = 'briaai/BRIA-3.0-ControlNet-Union'
79
 
80
  controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
81
  controlnet = BriaMultiControlNetModel([controlnet])
@@ -218,8 +218,8 @@ css="""
218
  with gr.Blocks(css=css) as demo:
219
  with gr.Column(elem_id="col-container"):
220
  gr.Markdown("""
221
- # BRIA-3.0-ControlNet-Union
222
- A unified ControlNet for BRIA-4B-Adapt model from Bria.ai. BRIA-4B-Adapt improve the generation of humans and illustrations compared to BRIA 2.3 while still trained on licensed data, and so provides full legal liability coverage for copyright and privacy infringement. Model card: [BRIA-3.0-ControlNet-Union](https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union). <br />
223
  """)
224
 
225
  with gr.Column():
 
75
  import PIL.Image as Image
76
 
77
  base_model = 'briaai/BRIA-4B-Adapt'
78
+ controlnet_model = 'briaai/BRIA-4B-Adapt-ControlNet-Union'
79
 
80
  controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
81
  controlnet = BriaMultiControlNetModel([controlnet])
 
218
  with gr.Blocks(css=css) as demo:
219
  with gr.Column(elem_id="col-container"):
220
  gr.Markdown("""
221
+ # BRIA-4B-Adapt-ControlNet-Union
222
+ A unified ControlNet for BRIA-4B-Adapt model from Bria.ai. BRIA-4B-Adapt improve the generation of humans and illustrations compared to BRIA 2.3 while still trained on licensed data, and so provides full legal liability coverage for copyright and privacy infringement. Model card: [BRIA-4B-Adapt-ControlNet-Union](https://huggingface.co/briaai/BRIA-4B-Adapt-ControlNet-Union). <br />
223
  """)
224
 
225
  with gr.Column():