merve HF Staff commited on
Commit
392a954
·
verified ·
1 Parent(s): 2685ee3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from typing import Union, Tuple
8
  from PIL import Image
9
 
10
  birefnet = AutoModelForImageSegmentation.from_pretrained(
11
- "ZhengPeng7/BiRefNet", low_cpu_mem_usage=False, trust_remote_code=True, torch_dtype=torch.float32, device_map=None
12
  )
13
  birefnet = birefnet.eval()
14
  #birefnet.to("cuda")
 
8
  from PIL import Image
9
 
10
  birefnet = AutoModelForImageSegmentation.from_pretrained(
11
+ "merve/BiRefNet", low_cpu_mem_usage=False, trust_remote_code=True, torch_dtype=torch.float32, device_map=None
12
  )
13
  birefnet = birefnet.eval()
14
  #birefnet.to("cuda")