kkzkkakky commited on
Commit
1ba71e9
·
verified ·
1 Parent(s): 4e94696

make secret for repo_id

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,6 +8,7 @@ import subprocess
8
 
9
  def load_private_code():
10
  token = os.environ.get("TOKEN", "")
 
11
  if token:
12
  try:
13
  subprocess.run(
@@ -21,7 +22,7 @@ def load_private_code():
21
  with TemporaryDirectory() as tmp_dir:
22
  # プライベートスペースからファイルをダウンロード
23
  api.snapshot_download(
24
- repo_id="neoai-inc-private/pj-andromeda",
25
  repo_type="space",
26
  local_dir=tmp_dir
27
  )
 
8
 
9
  def load_private_code():
10
  token = os.environ.get("TOKEN", "")
11
+ repo_id = os.environ.get("REPO_ID", "")
12
  if token:
13
  try:
14
  subprocess.run(
 
22
  with TemporaryDirectory() as tmp_dir:
23
  # プライベートスペースからファイルをダウンロード
24
  api.snapshot_download(
25
+ repo_id=repo_id,
26
  repo_type="space",
27
  local_dir=tmp_dir
28
  )