Spaces:
Running
Running
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 |
'''
|