Update app.py

app.py

@@ -289,11 +289,15 @@ def start_training(
         except Exception as e:
             print(f"Failed to add HF_TOKEN secret to new space: {e}")
 
-        toolkit_src = os.path.join(script_location, "ai-toolkit")
-        toolkit_dest = os.path.join(dataset_folder, "ai-toolkit")
-
-        if not os.path.exists(toolkit_dest):
-            shutil.copytree(toolkit_src, toolkit_dest)
+        import subprocess
+
+        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}",
+                shell=True,
+                check=True
+            )
 
         upload_folder(
             folder_path=dataset_folder,