Fix link
Browse files- download.py +1 -1
download.py
CHANGED
@@ -24,7 +24,7 @@ class OMG_download():
|
|
24 |
|
25 |
def download_model_sam(self):
|
26 |
REPO_ID = 'han-cai/efficientvit-sam'
|
27 |
-
filename_list = ['
|
28 |
if not os.path.exists('/home/user/app/checkpoint/sam/'):
|
29 |
os.makedirs('/home/user/app/checkpoint/sam/')
|
30 |
for filename in filename_list:
|
|
|
24 |
|
25 |
def download_model_sam(self):
|
26 |
REPO_ID = 'han-cai/efficientvit-sam'
|
27 |
+
filename_list = ['efficientvit_sam_xl1.pt']
|
28 |
if not os.path.exists('/home/user/app/checkpoint/sam/'):
|
29 |
os.makedirs('/home/user/app/checkpoint/sam/')
|
30 |
for filename in filename_list:
|