Hugging Face's logo

Spaces:
ZhengPeng7
/
BiRefNet_demo
Running on Zero

App Files Community
3
ZhengPeng7 commited on
Commit
7dd89ba
•
1 Parent(s): 4cf55f4

Update the well-trained weights.

Browse files
Files changed (2) hide show
  1. birefnet_dis.pth → BiRefNet_ep580.pth +1 -1
  2. app.py +1 -1
birefnet_dis.pth → BiRefNet_ep580.pth RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7db692b52f7f855c41d05e4427a05ac63a755f39f80e11d6185eb48b80acbea8
3
  size 848968257
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b4510f31d72e41507a4b75c4e62206b1d7e2223e0125b29644acd4b142793b0
3
  size 848968257
app.py CHANGED
@@ -29,7 +29,7 @@ class ImagePreprocessor():
29
 
30
 
31
  model = BiRefNet().to(device)
32
- state_dict = './birefnet_dis.pth'
33
  if os.path.exists(state_dict):
34
  birefnet_dict = torch.load(state_dict, map_location=device)
35
  unwanted_prefix = '_orig_mod.'
 
29
 
30
 
31
  model = BiRefNet().to(device)
32
+ state_dict = './BiRefNet_ep580.pth'
33
  if os.path.exists(state_dict):
34
  birefnet_dict = torch.load(state_dict, map_location=device)
35
  unwanted_prefix = '_orig_mod.'