Update app_3.py

app_3.py

@@ -158,7 +158,7 @@ from transformers import AutoModelForImageSegmentation
 rmbg = AutoModelForImageSegmentation.from_pretrained(
             "ZhengPeng7/BiRefNet", trust_remote_code=True
         )
-rmbg = rmbg.to_device(device=device, dtype=torch.float32) 
+rmbg = rmbg.to(device=device, dtype=torch.float32) 
 
 # remove bg
 # rmbg = AutoModelForImageSegmentation.from_pretrained(