Spaces:
Sleeping
Sleeping
Update TryYours-Virtual-Try-On/posenet/converter/wget.py
Browse files
TryYours-Virtual-Try-On/posenet/converter/wget.py
CHANGED
@@ -25,7 +25,7 @@ def download_file(checkpoint, filename, base_dir):
|
|
25 |
f.write(response.content)
|
26 |
f.close()
|
27 |
|
28 |
-
def download(checkpoint, base_dir='
|
29 |
save_dir = os.path.join(base_dir, checkpoint)
|
30 |
if not os.path.exists(save_dir):
|
31 |
os.makedirs(save_dir)
|
|
|
25 |
f.write(response.content)
|
26 |
f.close()
|
27 |
|
28 |
+
def download(checkpoint, base_dir='TryYours-Virtual-Try-On/posenet/converter/weights/'):
|
29 |
save_dir = os.path.join(base_dir, checkpoint)
|
30 |
if not os.path.exists(save_dir):
|
31 |
os.makedirs(save_dir)
|