ttttdiva commited on
Commit
8a73292
·
verified ·
1 Parent(s): 4d303e2

Upload main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -11
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
- if model_hf_url in model_list:
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
- # 例: https://huggingface.co/REPO_ID/tree/main/<暗号化後トップフォルダ>
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: