Update app.py

app.py

@@ -289,6 +289,12 @@ 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)
+
         upload_folder(
             folder_path=dataset_folder,
             repo_id=new_space_id,