Update app.py

app.py

@@ -165,16 +165,17 @@ def resume_training(extra_steps, push_to_hub, repo_id, run_dir_text):
         "lerobot-train "
         f"--output_dir='{run_dir}' "
         "--resume=true "
+        f"--config_path='{cfg_path}' "
         f"--steps={extra_steps} "
         "--eval_freq=500 "
         "--save_freq=500 "
+        f"{push_flags}"
     )
     if os.path.exists(cfg_path):
         base += f"--config_path='{cfg_path}' "
     else:
-        # Fallback to minimal schema (prevents policy parsing errors)
         base += "--policy.type=diffusion --dataset.repo_id=lerobot/pusht --env.type=pusht "
-
+    
     cmd = base + push_flags
     rc, tail = _run(cmd, log)
     msg = f"Resumed run at: {run_dir}\nResume exited rc={rc}\n\n=== train.log tail ===\n{tail}"