Hugging Face's logo

Spaces:
briaai
/
BRIA-RMBG-2.0
Running on Zero

App Files Community
8
OriLib commited on
Commit
20dbe2a
·
verified ·
1 Parent(s): 1afdc8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from torchvision import transforms
9
  torch.set_float32_matmul_precision(["high", "highest"][0])
10
 
11
  birefnet = AutoModelForImageSegmentation.from_pretrained(
12
- "ZhengPeng7/BiRefNet", trust_remote_code=True
13
  )
14
  birefnet.to("cuda")
15
  transform_image = transforms.Compose(
 
9
  torch.set_float32_matmul_precision(["high", "highest"][0])
10
 
11
  birefnet = AutoModelForImageSegmentation.from_pretrained(
12
+ "briaai/BRIA-RMBG-1.4", trust_remote_code=True
13
  )
14
  birefnet.to("cuda")
15
  transform_image = transforms.Compose(