Spaces:
Running
Running
fix official weights
Browse filesThe model is no longer available from the link
app.py
CHANGED
@@ -25,8 +25,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
25 |
# Download official weights
|
26 |
if not os.path.exists("saved_models"):
|
27 |
os.mkdir("saved_models")
|
28 |
-
|
29 |
-
gdown.download(MODEL_PATH_URL, "saved_models/isnet.pth", use_cookies=False)
|
30 |
|
31 |
class GOSNormalize(object):
|
32 |
'''
|
|
|
25 |
# Download official weights
|
26 |
if not os.path.exists("saved_models"):
|
27 |
os.mkdir("saved_models")
|
28 |
+
os.system("mv isnet.pth saved_models/")
|
|
|
29 |
|
30 |
class GOSNormalize(object):
|
31 |
'''
|