DevWild commited on
Commit
e9f2c7e
·
verified ·
1 Parent(s): 729ea31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -284,6 +284,7 @@ def start_training(
284
  try:
285
  # Inject HF_TOKEN into the new space
286
  api.add_space_secret(repo_id=new_space_id, key="HF_TOKEN", value=hf_token)
 
287
  print(f"Added HF_TOKEN secret to {new_space_id}")
288
  except Exception as e:
289
  print(f"Failed to add HF_TOKEN secret to new space: {e}")
 
284
  try:
285
  # Inject HF_TOKEN into the new space
286
  api.add_space_secret(repo_id=new_space_id, key="HF_TOKEN", value=hf_token)
287
+ api.add_space_secret(repo_id=new_space_id, key="HF_DATASET", value=destination_folder)
288
  print(f"Added HF_TOKEN secret to {new_space_id}")
289
  except Exception as e:
290
  print(f"Failed to add HF_TOKEN secret to new space: {e}")