Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -25,8 +25,9 @@ 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 |
-
MODEL_PATH_URL = "https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn"
|
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 |
+
# MODEL_PATH_URL = "https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn"
|
29 |
+
# gdown.download(MODEL_PATH_URL, "saved_models/isnet.pth", use_cookies=False)
|
30 |
+
os.system("mv isnet.pth saved_models/")
|
31 |
|
32 |
class GOSNormalize(object):
|
33 |
'''
|