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

fix: if it's None

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,10 +11,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(
 
11
  TARGET = os.getenv("TARGET")
12
 
13
  def func(hf_token, repo, revision, target_repo):
14
+ hf_token = hf_token if hf_token is None else HF_TOKEN
15
+ repo = repo if repo is None else REPO
16
+ revision = revision if revision is None else REVISION
17
+ target_repo = target_repo if target_repo is None else TARGET
18
 
19
  print('download the desired model locally ...')
20
  hf_api.snapshot_download(