Upload temp.py with huggingface_hub

temp.py

@@ -17,7 +17,7 @@
 
 
 import os
-from huggingface_hub import HfApi, upload_file
+from huggingface_hub import upload_file
 from tqdm import tqdm
 
 # Hugging Face 토큰을 환경 변수로 설정
@@ -27,22 +27,16 @@ hf_token = os.getenv("HF_TOKEN")
 folder_path = "."  # 현재 디렉토리
 repo_id = "yerang/txt2liveportrait"  # 리포지토리 ID
 
-# Hugging Face API 인스턴스 생성
-api = HfApi()
+# 제외할 파일 크기 (단위: 바이트, 여기서는 100MB 이상인 파일을 제외)
+max_file_size = 100 * 1024 * 1024  # 100MB
 
-# 제외할 폴더 및 파일 패턴 설정
-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)
+        full_path = os.path.join(root, file)
+        # 파일 크기 확인
+        if os.path.getsize(full_path) < max_file_size:  # 설정한 크기 이하인 파일만 추가
             rel_path_in_repo = os.path.relpath(full_path, folder_path)
             file_paths.append((full_path, rel_path_in_repo))