vpcom commited on
Commit
a961aec
·
1 Parent(s): 1b49c62

fix: use default environment variables when user don't provide any inputs

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -8,13 +8,19 @@ CURRENT_MODEL_PATH = "./current_model"
8
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
9
  REPO = os.getenv("REPO")
10
  REVISION = os.getenv("REVISION")
 
11
 
12
  def func(hf_token, repo, revision, target_repo):
 
 
 
 
 
13
  print('download the desired model locally ...')
14
  hf_api.snapshot_download(
15
- repo_id=repo if len(repo)>0 else REPO,
16
- revision=revision if len(revision)>0 else REVISION,
17
- token=hf_token if len(hf_token)>0 else HF_TOKEN,
18
  local_dir=CURRENT_MODEL_PATH,
19
  )
20
  print(os.listdir(CURRENT_MODEL_PATH))
 
8
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
9
  REPO = os.getenv("REPO")
10
  REVISION = os.getenv("REVISION")
11
+ TARGET = os.getenv("TARGET")
12
 
13
  def func(hf_token, repo, revision, target_repo):
14
+ hf_token = hf_token if len(hf_token)>0 else HF_TOKEN
15
+ repo = repo if len(repo)>0 else REPO
16
+ revision = revision if len(revision)>0 else REVISION
17
+ target_repo = target_repo if len(target_repo)>0 else TARGET
18
+
19
  print('download the desired model locally ...')
20
  hf_api.snapshot_download(
21
+ repo_id=repo,
22
+ revision=revision,
23
+ token=hf_token,
24
  local_dir=CURRENT_MODEL_PATH,
25
  )
26
  print(os.listdir(CURRENT_MODEL_PATH))