Spaces:
Sleeping
Sleeping
fix: use default environment variables when user don't provide any inputs
Browse files
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
|
16 |
-
revision=revision
|
17 |
-
token=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))
|