Update app.py

app.py

@@ -257,6 +257,9 @@ def start_training(
         shutil.copy(script_location + "/script.py", dataset_folder)
         # copy requirements.autotrain to dataset_folder as requirements.txt
         shutil.copy(script_location + "/requirements.autotrain", dataset_folder + "/requirements.txt")
+        # Copy additional runtime files
+        shutil.copy(script_location + "/Dockerfile", dataset_folder)
+        shutil.copy(script_location + "/space.yaml", dataset_folder)
         
         # command to run autotrain spacerunner
         # --- Pre-create subspace and inject token ---