SVDiff-Training-UI / uploader.py
mkshing
first commit
e2a20af
from __future__ import annotations
from huggingface_hub import HfApi
class Uploader:
def __init__(self, hf_token: str | None):
self.api = HfApi(token=hf_token)
def get_username(self) -> str:
return self.api.whoami()['name']
def upload(self,
folder_path: str,
repo_name: str,
organization: str = '',
repo_type: str = 'model',
private: bool = True,
delete_existing_repo: bool = False) -> str:
if not folder_path:
raise ValueError
if not repo_name:
raise ValueError
if not organization:
organization = self.get_username()
repo_id = f'{organization}/{repo_name}'
if delete_existing_repo:
try:
self.api.delete_repo(repo_id, repo_type=repo_type)
except Exception:
pass
try:
self.api.create_repo(repo_id, repo_type=repo_type, private=private)
self.api.upload_folder(repo_id=repo_id,
folder_path=folder_path,
path_in_repo='.',
repo_type=repo_type)
url = f'https://huggingface.co/{repo_id}'
message = f'Your model was successfully uploaded to <a href="{url}" target="_blank">{url}</a>.'
except Exception as e:
message = str(e)
return message