Fucius commited on
Commit
72c5c13
1 Parent(s): 947a4f8

Update download.py

Browse files
Files changed (1) hide show
  1. download.py +4 -4
download.py CHANGED
@@ -25,13 +25,13 @@ class OMG_download():
25
  def download_model_sam(self):
26
  REPO_ID = 'han-cai/efficientvit-sam'
27
  filename_list = ['xl1.pt']
28
- if not os.path.exists('./checkpoints/sam/'):
29
- os.makedirs('./checkpoints/sam/')
30
  for filename in filename_list:
31
- local_file = os.path.join('./checkpoints/sam/', filename)
32
 
33
  if not os.path.exists(local_file):
34
- hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/sam/', local_dir_use_symlinks=False)
35
 
36
  def download_model_yoloworld(self):
37
  REPO_ID = 'Fucius/OMG'
 
25
  def download_model_sam(self):
26
  REPO_ID = 'han-cai/efficientvit-sam'
27
  filename_list = ['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:
31
+ local_file = os.path.join('/home/user/app/checkpoint/sam/', filename)
32
 
33
  if not os.path.exists(local_file):
34
+ hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='/home/user/app/checkpoint/sam/', local_dir_use_symlinks=False)
35
 
36
  def download_model_yoloworld(self):
37
  REPO_ID = 'Fucius/OMG'