Update app.py

app.py

@@ -294,7 +294,7 @@ def start_training(
         toolkit_src = os.path.join(dataset_folder, "ai-toolkit")
         if not os.path.exists(toolkit_src):
             subprocess.run(
-                f"git clone https://github.com/ostris/ai-toolkit.git {toolkit_src}",
+                f"git clone https://github.com/ostris/ai-toolkit.git ai-toolkit && cd ai-toolkit && git submodule update --init --recursive {toolkit_src}",
                 shell=True,
                 check=True
             )
@@ -306,7 +306,15 @@ def start_training(
             token=hf_token,
             commit_message="Pushing training files from controller"
         )
-            
+
+        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}")
+
+        
         cmd = f"autotrain spacerunner --project-name {slugged_lora_name} --script-path {dataset_folder}"
         cmd += f" --username {profile.username} --token {hf_token} --backend spaces-l4x1"
         outcome = subprocess.run(cmd.split(), capture_output=True, text=True)