DevWild commited on
Commit
db66f76
·
verified ·
1 Parent(s): af9cbc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -284,7 +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
- api.add_space_secret(repo_id=new_space_id, key="HF_DATASET", value=dataset_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}")
 
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=f"{profile.username}/autotrain-{slugged_lora_name}")
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}")