Upload hfhub.py with huggingface_hub

hfhub.py

@@ -1,7 +1,17 @@
 import os
 from pathlib import Path
 
-from huggingface_hub import HfApi, Repository, hf_hub_download, snapshot_download
+from huggingface_hub import (
+    HfApi,
+    Repository,
+    create_repo,
+    get_full_repo_name,
+    hf_hub_download,
+    login,
+    logout,
+    snapshot_download,
+    upload_file,
+)
 
 repo_name = "test2"
 repo_dir = Path("hfhub")
@@ -10,16 +20,18 @@ download_checkpoint_with_name = "test_checkpoint"
 
 (Path(repo_dir) / "checkpoints").mkdir(parents=True, exist_ok=True)
 
+login(token=token)
 hf_api = HfApi(token=token)
 
 # username/repo_name
-repo_name = hf_api.get_full_repo_name(repo_name)
+repo_name = get_full_repo_name(repo_name)
 
-repo_url = hf_api.create_repo(
+repo_url = create_repo(
     repo_id=repo_name, repo_type="model", exist_ok=True, private=True
 )
 
-hf_api.upload_file(
+
+upload_file(
     repo_id=repo_name,
     repo_type="model",
     path_or_fileobj="hfhub.py",
@@ -43,5 +55,6 @@ snapshot_download(
     resume_download=True,
     local_dir=Path(repo_dir) / "datasets",
     allow_patterns="splits_gulp_rgb/*",
-    token=token,
 )
+
+logout()