Update restore.py

restore.py

@@ -23,7 +23,7 @@ os.makedirs(env["TMPDIR"], exist_ok=True)
 # Step 1: List repo files
 print("[Restore] Listing backups in repo...")
 result = subprocess.run(
-    ["huggingface-cli", "ls-files", BACKUP_REPO, "--repo-type", "dataset"],
+    ["hf", "ls-files", BACKUP_REPO, "--repo-type", "dataset"],
     stdout=subprocess.PIPE,
     stderr=subprocess.STDOUT,
     text=True,
@@ -43,7 +43,7 @@ print(f"[Restore] Found latest backup: {latest}")
 # Step 3: Download
 tmpfile = os.path.join(tempfile.gettempdir(), "restore.tar.gz")
 subprocess.run(
-    ["huggingface-cli", "download", BACKUP_REPO, latest, "--repo-type", "dataset", "--local-dir", "/tmp/restore", "--force-download"],
+    ["hf", "download", BACKUP_REPO, latest, "--repo-type", "dataset", "--local-dir", "/tmp/restore", "--force"],
     check=True,
     env=env,
 )