Update app_3.py

app_3.py

@@ -150,14 +150,14 @@ 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/RMBG-1.4", trust_remote_code=True)
-# rmbg = rmbg.to(device=device, dtype=torch.float32)  # Keep this as float32
+rmbg = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4", trust_remote_code=True)
+rmbg = rmbg.to(device=device, dtype=torch.float32)  # Keep this as float32
 
 # remove bg
-rmbg = AutoModelForImageSegmentation.from_pretrained(
-    "ZhengPeng7/BiRefNet", trust_remote_code=True
-)
-rmbg = rmbg.to(device)
+# rmbg = AutoModelForImageSegmentation.from_pretrained(
+#     "ZhengPeng7/BiRefNet", trust_remote_code=True
+# )
+# rmbg = rmbg.to(device)
 
 model = DepthAnythingV2(encoder='vits', features=64, out_channels=[48, 96, 192, 384])
 model.load_state_dict(torch.load('checkpoints/depth_anything_v2_vits.pth', map_location=device))
@@ -783,19 +783,12 @@ def process_relight_bg(input_fg, input_bg, prompt, image_width, image_height, nu
 
 quick_prompts = [
     'sunshine from window',
-    'neon light, city',
-    'sunset over sea',
     'golden time',
-    'sci-fi RGB glowing, cyberpunk',
     'natural lighting',
     'warm atmosphere, at home, bedroom',
-    'magic lit',
-    'evil, gothic, Yharnam',
-    'light and shadow',
     'shadow from window',
     'soft studio lighting',
     'home atmosphere, cozy bedroom illumination',
-    'neon, Wong Kar-wai, warm'
 ]
 quick_prompts = [[x] for x in quick_prompts]