Update app.py

app.py

@@ -216,7 +216,7 @@ def start_training(
     config["config"]["process"][0]["network"]["linear_alpha"] = int(rank)
     config["config"]["process"][0]["datasets"][0]["folder_path"] = dataset_folder
     config["config"]["process"][0]["save"]["push_to_hub"] = True
-    config["config"]["process"][0]["save"]["hf_repo_id"] = model_repo_id
+    #config["config"]["process"][0]["save"]["hf_repo_id"] = model_repo_id
     try:
         username = whoami()["name"] if not is_spaces else profile.username
     except:
@@ -294,6 +294,7 @@ def start_training(
             # Inject HF_TOKEN into the new space
             api.add_space_secret(repo_id=new_space_id, key="HF_TOKEN", value=hf_token)
             api.add_space_secret(repo_id=new_space_id, key="HF_DATASET", value=f"{profile.username}/autotrain-{slugged_lora_name}")
+            api.add_space_secret(repo_id=new_space_id, key="MODEL_REPO_ID", value=model_repo_id)
             print(f"Added HF_TOKEN secret to {new_space_id}")
         except Exception as e:
             print(f"Failed to add HF_TOKEN secret to new space: {e}")