Update app.py

app.py

@@ -80,7 +80,7 @@ def get_size(init_image):
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load, init model
-controlnet = ControlNetModel().from_pretrained("briaai/BRIA-2.3-ControlNet-GenFill", torch_dtype=torch.float16)
+controlnet = ControlNetModel().from_pretrained("briaai/BRIA-2.3-ControlNet-GenFill-BETA", torch_dtype=torch.float16)
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained("briaai/BRIA-2.3", controlnet=controlnet.to(dtype=torch.float16), torch_dtype=torch.float16, vae=vae) #force_zeros_for_empty_prompt=False, # vae=vae)
 
@@ -207,7 +207,7 @@ with image_blocks as demo:
         gr.HTML('''
           <p style="margin-bottom: 10px; font-size: 94%">
             This is a demo for 
-            <a href="https://huggingface.co/briaai/BRIA-2.3-ControlNet-GenFill" target="_blank">BRIA 2.3 ControlNet GenFill</a>. 
+            <a href="https://huggingface.co/briaai/BRIA-2.3-ControlNet-GenFill-BETA" target="_blank">BRIA 2.3 ControlNet GenFill</a>. 
             BRIA GenFill focuses on the inpainting tasks Modify, Add, Replace and trained on licensed data, and provides full legal liability coverage for copyright and privacy infringement.
           </p>
         ''')