File size: 1,987 Bytes
d53425e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
57c6678
a9138c3
d53425e
57c6678
d53425e
57c6678
d53425e
 
 
 
 
 
 
 
a9138c3
 
 
 
 
d53425e
 
a9138c3
 
 
d53425e
a9138c3
 
 
 
d53425e
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
# import os
# from huggingface_hub import upload_folder

# # Hugging Face ํ† ํฐ์„ ํ™˜๊ฒฝ ๋ณ€์ˆ˜๋กœ ์„ค์ •
# hf_token = os.getenv("HF_TOKEN")

# # ํ˜„์žฌ ๋””๋ ‰ํ† ๋ฆฌ ์ „์ฒด๋ฅผ ์—…๋กœ๋“œ
# folder_path = "."  # ํ˜„์žฌ ๋””๋ ‰ํ† ๋ฆฌ

# # Hugging Face ๋ฆฌํฌ์ง€ํ† ๋ฆฌ์— ํด๋” ์—…๋กœ๋“œ
# upload_folder(
#     folder_path=folder_path,  # ์—…๋กœ๋“œํ•  ํด๋” ๊ฒฝ๋กœ
#     path_in_repo="",          # ๋ฆฌํฌ์ง€ํ† ๋ฆฌ ๋‚ด ๊ฒฝ๋กœ (๋ฃจํŠธ๋กœ ์„ค์ •)
#     repo_id="yerang/txt2liveportrait",  # ๋ฆฌํฌ์ง€ํ† ๋ฆฌ ID
#     token=hf_token            # ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ ํ† ํฐ ์‚ฌ์šฉ
# )


import os
from huggingface_hub import HfApi, upload_file
from tqdm import tqdm

# Hugging Face ํ† ํฐ์„ ํ™˜๊ฒฝ ๋ณ€์ˆ˜๋กœ ์„ค์ •
hf_token = os.getenv("HF_TOKEN")

# ํ˜„์žฌ ๋””๋ ‰ํ† ๋ฆฌ ์ „์ฒด๋ฅผ ์—…๋กœ๋“œ
folder_path = "."  # ํ˜„์žฌ ๋””๋ ‰ํ† ๋ฆฌ
repo_id = "yerang/txt2liveportrait"  # ๋ฆฌํฌ์ง€ํ† ๋ฆฌ ID

# Hugging Face API ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ
api = HfApi()

# ์ œ์™ธํ•  ํด๋” ๋ฐ ํŒŒ์ผ ํŒจํ„ด ์„ค์ •
exclude_dirs = {".git", ".ipynb_checkpoints"}
exclude_files = {".gitignore", ".gitattributes", ".gitmodules"}

# ํŒŒ์ผ ๋ชฉ๋ก ๊ฐ€์ ธ์˜ค๊ธฐ (ํ•„ํ„ฐ๋ง ํฌํ•จ)
file_paths = []
for root, dirs, files in os.walk(folder_path):
    # ์ œ์™ธํ•  ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ํƒ์ƒ‰ ๋Œ€์ƒ์—์„œ ์ œ์™ธ
    dirs[:] = [d for d in dirs if d not in exclude_dirs]

    for file in files:
        if file not in exclude_files:  # ์ œ์™ธํ•  ํŒŒ์ผ ํ•„ํ„ฐ๋ง
            full_path = os.path.join(root, file)
            rel_path_in_repo = os.path.relpath(full_path, folder_path)
            file_paths.append((full_path, rel_path_in_repo))

# ํŒŒ์ผ ์—…๋กœ๋“œ ์ง„ํ–‰ ์ƒํ™ฉ ํ‘œ์‹œ
for full_path, rel_path_in_repo in tqdm(file_paths, desc="Uploading files"):
    upload_file(
        path_or_fileobj=full_path,     # ์‹ค์ œ ํŒŒ์ผ ๊ฒฝ๋กœ
        path_in_repo=rel_path_in_repo,  # ๋ฆฌํฌ์ง€ํ† ๋ฆฌ ๋‚ด ๊ฒฝ๋กœ
        repo_id=repo_id,               # ๋ฆฌํฌ์ง€ํ† ๋ฆฌ ID
        token=hf_token                 # Hugging Face ํ† ํฐ
    )