rikunarita-2 commited on
Commit
e2cb84b
·
verified ·
1 Parent(s): 4ada331

Create uploader.py

Browse files
Files changed (1) hide show
  1. backend/uploader.py +16 -0
backend/uploader.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from huggingface_hub import HfApi, create_repo, upload_folder
2
+ import os
3
+
4
+ def upload_to_hf(folder_path: str, repo_name: str, token: str):
5
+ api = HfApi()
6
+ user = api.whoami(token)["name"]
7
+ repo_id = f"{user}/{repo_name}"
8
+ create_repo(repo_id, token=token, exist_ok=True)
9
+ upload_folder(
10
+ folder_path=folder_path,
11
+ repo_id=repo_id,
12
+ repo_type="model",
13
+ token=token,
14
+ commit_message="Evolutionary merge upload"
15
+ )
16
+ return repo_id