ttttdiva commited on
Commit
aff6f73
·
verified ·
1 Parent(s): 0a8e4bf

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -4
main.py CHANGED
@@ -430,11 +430,10 @@ class CivitAICrawler:
430
  encrypted_top_name = self.upload_folder_encrypted(folder)
431
  shutil.rmtree(folder, ignore_errors=True)
432
 
433
- # ここで、暗号化後のトップフォルダ名をURLに含める
434
- model_hf_url = f"https://huggingface.co/{self.repo_ids['current']}/tree/main/{encrypted_top_name}"
435
 
436
- # あるいは、さらにサブフォルダ構造があるなら適宜URLを調整
437
- self.append_model_list(model_id, model_hf_url)
438
 
439
  except Exception as e:
440
  logger.error("Unexpected error in process_model(%s): %s", model_url, e)
 
430
  encrypted_top_name = self.upload_folder_encrypted(folder)
431
  shutil.rmtree(folder, ignore_errors=True)
432
 
433
+ modelpage_name = model_info.get("name", "Unnamed Model")
 
434
 
435
+ model_hf_url = f"https://huggingface.co/{self.repo_ids['current']}/tree/main/{encrypted_top_name}"
436
+ self.append_model_list(modelpage_name, model_hf_url)
437
 
438
  except Exception as e:
439
  logger.error("Unexpected error in process_model(%s): %s", model_url, e)