Update app_3.py

app_3.py

@@ -52,6 +52,7 @@ WIDTH = 768
 MAX_SEED = np.iinfo(np.int32).max
 
 
+
 import supervision as sv
 import torch
 from PIL import Image
@@ -150,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)
+rmbg = AutoModelForImageSegmentation.from_pretrained("briaai/BRIA-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