Update app.py

app.py

@@ -195,17 +195,6 @@ def start_training(
     print("Started training")
     slugged_lora_name = slugify(lora_name)
 
-    # Create token
-    hf_token = oauth_token.token
-    api = HfApi(token=hf_token)
-    
-    # Create model repo
-    model_repo_id = f"{profile.username}/{slugged_lora_name}"
-    try:
-        api.create_repo(repo_id=model_repo_id, repo_type="model", private=True)
-        print(f"Created model repo {model_repo_id}")
-    except Exception as e:
-        print(f"Model repo might already exist: {e}")
 #    try:
 #        api.create_repo(repo_id=model_repo_id, repo_type="model", private=True)
 #        print(f"Created model repo {model_repo_id}")
@@ -285,6 +274,19 @@ def start_training(
         shutil.copy(script_location + "/space.yaml", dataset_folder)
         
         # command to run autotrain spacerunner
+
+        # Create token
+        hf_token = oauth_token.token
+        api = HfApi(token=hf_token)
+    
+        # Create model repo
+        model_repo_id = f"{profile.username}/{slugged_lora_name}"
+        try:
+            api.create_repo(repo_id=model_repo_id, repo_type="model", private=True)
+            print(f"Created model repo {model_repo_id}")
+        except Exception as e:
+            print(f"Model repo might already exist: {e}")
+        
         # --- Pre-create subspace and inject token ---
         new_space_id = f"{profile.username}/autotrain-{slugged_lora_name}"