Harshil748 commited on
Commit
989343f
Β·
1 Parent(s): b1de92d

Improve model download logging

Browse files
Files changed (1) hide show
  1. download_models.py +13 -3
download_models.py CHANGED
@@ -13,31 +13,41 @@ MODEL_REPO = "Harshil748/VoiceAPI-Models"
13
 
14
  def download_all_models():
15
  """Download all models from HuggingFace."""
 
16
  print("πŸš€ Starting model downloads...")
17
  print(f" Source: {MODEL_REPO}")
18
  print(f" Target: {MODELS_DIR.absolute()}")
 
19
 
20
  MODELS_DIR.mkdir(exist_ok=True)
21
 
22
  try:
23
- print("\nοΏ½οΏ½ Downloading all models...")
24
  snapshot_download(
25
  repo_id=MODEL_REPO,
26
  local_dir=MODELS_DIR,
27
  local_dir_use_symlinks=False,
28
  ignore_patterns=["*.md", ".gitattributes"],
29
  )
30
- print("βœ… All models downloaded successfully!")
31
 
32
  # List downloaded voices
33
  print("\nπŸ“¦ Downloaded voices:")
 
34
  for voice in sorted(MODELS_DIR.iterdir()):
35
  if voice.is_dir():
36
  files = list(voice.glob("*"))
37
- print(f" βœ“ {voice.name}: {len(files)} files")
 
 
 
 
 
38
 
39
  except Exception as e:
40
  print(f"❌ Failed to download models: {e}")
 
 
41
  raise
42
 
43
 
 
13
 
14
  def download_all_models():
15
  """Download all models from HuggingFace."""
16
+ print("=" * 60)
17
  print("πŸš€ Starting model downloads...")
18
  print(f" Source: {MODEL_REPO}")
19
  print(f" Target: {MODELS_DIR.absolute()}")
20
+ print("=" * 60)
21
 
22
  MODELS_DIR.mkdir(exist_ok=True)
23
 
24
  try:
25
+ print("\nπŸ“₯ Downloading all models from HuggingFace...")
26
  snapshot_download(
27
  repo_id=MODEL_REPO,
28
  local_dir=MODELS_DIR,
29
  local_dir_use_symlinks=False,
30
  ignore_patterns=["*.md", ".gitattributes"],
31
  )
32
+ print("\nβœ… All models downloaded successfully!")
33
 
34
  # List downloaded voices
35
  print("\nπŸ“¦ Downloaded voices:")
36
+ total_size = 0
37
  for voice in sorted(MODELS_DIR.iterdir()):
38
  if voice.is_dir():
39
  files = list(voice.glob("*"))
40
+ size = sum(f.stat().st_size for f in files if f.is_file())
41
+ total_size += size
42
+ print(f" βœ“ {voice.name}: {len(files)} files ({size / 1024 / 1024:.1f} MB)")
43
+
44
+ print(f"\nπŸ“Š Total size: {total_size / 1024 / 1024 / 1024:.2f} GB")
45
+ print("=" * 60)
46
 
47
  except Exception as e:
48
  print(f"❌ Failed to download models: {e}")
49
+ import traceback
50
+ traceback.print_exc()
51
  raise
52
 
53