Update app_3.py

app_3.py

@@ -151,7 +151,7 @@ vae = AutoencoderKL.from_pretrained(sd15_name, subfolder="vae")
 unet = UNet2DConditionModel.from_pretrained(sd15_name, subfolder="unet")
 # Load model directly
 from transformers import AutoModelForImageSegmentation
-rmbg = AutoModelForImageSegmentation.from_pretrained("briaai/BRIA-RMBG-2.0", trust_remote_code=True, token=os.getenv('token'))
+rmbg = AutoModelForImageSegmentation.from_pretrained("RMBG-2.0", trust_remote_code=True)#, token=os.getenv('token'))
 rmbg = rmbg.to(device=device, dtype=torch.float32)  # Keep this as float32
 
 # remove bg