Spaces:
Running
Running
Update app.py
Browse files
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=
|
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}")
|