Create push_to_hub.py

push_to_hub.py

@@ -0,0 +1,56 @@
+"""Push hv-core to the Hugging Face Hub."""
+import os
+
+from huggingface_hub import HfApi, create_repo
+
+from hv_core import HVCore
+
+REPO_ID = "zeechimp/hv-core"
+
+
+def main() -> None:
+    api = HfApi()
+
+    # Create the repo if it doesn't exist
+    create_repo(REPO_ID, repo_type="model", exist_ok=True)
+
+    # Save model config locally
+    model = HVCore()
+    model.save_pretrained("./hv_core_model")
+
+    # Upload the saved config folder
+    api.upload_folder(
+        folder_path="./hv_core_model",
+        repo_id=REPO_ID,
+        path_in_repo=".",
+        commit_message="Upload hv-core config",
+    )
+
+    # Upload the source + supporting files
+    for fname in [
+        "hv_core.py",
+        "app.py",
+        "example.py",
+        "push_to_hub.py",
+        "README.md",
+        "requirements.txt",
+        "LICENSE",
+        ".gitattributes",
+    ]:
+        if not os.path.exists(fname):
+            print(f"  skip (missing): {fname}")
+            continue
+        api.upload_file(
+            path_or_fileobj=fname,
+            path_in_repo=fname,
+            repo_id=REPO_ID,
+            commit_message=f"Upload {fname}",
+        )
+        print(f"  uploaded: {fname}")
+
+    print()
+    print(f"pushed to https://huggingface.co/{REPO_ID}")
+
+
+if __name__ == "__main__":
+    main()