Spaces:
Build error
Build error
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))
|