Upload main.py
Browse files
main.py
CHANGED
@@ -527,16 +527,9 @@ class CivitAICrawler:
|
|
527 |
|
528 |
os.makedirs(folder, exist_ok=True)
|
529 |
|
530 |
-
# すでに同フォルダURLがあるかどうか
|
531 |
model_list = self.read_model_list()
|
532 |
|
533 |
-
|
534 |
-
repo_id = self.repo_ids['current']
|
535 |
-
repo_files = self.get_repo_info(repo_id)
|
536 |
-
old_versions_files = [f for f in repo_files if f.startswith(f"{folder}/old_versions/")]
|
537 |
-
existing_old_version_files = [os.path.basename(f) for f in old_versions_files]
|
538 |
-
else:
|
539 |
-
existing_old_version_files = []
|
540 |
|
541 |
self.download_model(model_info["modelVersions"], folder, existing_old_version_files)
|
542 |
self.download_images(model_info["modelVersions"], folder)
|
@@ -544,13 +537,11 @@ class CivitAICrawler:
|
|
544 |
self.save_model_info(model_info, folder)
|
545 |
|
546 |
# ========== rclone で暗号化フォルダをアップロード ==========
|
547 |
-
# ここで「実際にHF上に作成された暗号化後のトップフォルダ名」を取得
|
548 |
encrypted_top_name = self.upload_folder_encrypted(folder)
|
549 |
|
550 |
# モデルリスト更新
|
551 |
-
# ここで実際の暗号化フォルダ名を使ったURLを書き込む
|
552 |
modelpage_name = model_info.get("name", "Unnamed Model")
|
553 |
-
#
|
554 |
model_hf_url = f"https://huggingface.co/{self.repo_ids['current']}/tree/main/{encrypted_top_name}"
|
555 |
|
556 |
with open(self.config.LIST_FILE, "a", encoding="utf-8") as f:
|
|
|
527 |
|
528 |
os.makedirs(folder, exist_ok=True)
|
529 |
|
|
|
530 |
model_list = self.read_model_list()
|
531 |
|
532 |
+
existing_old_version_files = []
|
|
|
|
|
|
|
|
|
|
|
|
|
533 |
|
534 |
self.download_model(model_info["modelVersions"], folder, existing_old_version_files)
|
535 |
self.download_images(model_info["modelVersions"], folder)
|
|
|
537 |
self.save_model_info(model_info, folder)
|
538 |
|
539 |
# ========== rclone で暗号化フォルダをアップロード ==========
|
|
|
540 |
encrypted_top_name = self.upload_folder_encrypted(folder)
|
541 |
|
542 |
# モデルリスト更新
|
|
|
543 |
modelpage_name = model_info.get("name", "Unnamed Model")
|
544 |
+
# 暗号化されたフォルダを使った有効なURL
|
545 |
model_hf_url = f"https://huggingface.co/{self.repo_ids['current']}/tree/main/{encrypted_top_name}"
|
546 |
|
547 |
with open(self.config.LIST_FILE, "a", encoding="utf-8") as f:
|