Spaces:
Running
Running
Commit
Β·
989343f
1
Parent(s):
b1de92d
Improve model download logging
Browse files- 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
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|