Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
7dd89ba
1
Parent(s):
4cf55f4
Update the well-trained weights.
Browse files- birefnet_dis.pth → BiRefNet_ep580.pth +1 -1
- 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:
|
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 = './
|
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.'
|