test.py

import sys
sys.path.append("src")
from model_loader import ModelLoader

# Load from HuggingFace Hub (when available)
loader = ModelLoader.from_pretrained("adrianstanea/Ro-Matcha-TTS")

# Or load from local path
loader = ModelLoader.from_pretrained("./")

# List available models
print(loader.list_models())
# {'swara': {...}, 'bas_10': {...}, 'bas_950': {...}, ...}

# Load production-ready BAS speaker
model_info = loader.load_models(model="bas_950")
print(f"Model: {model_info['model_name']}")
print(f"Path: {model_info['model_path']}")

# Load few-shot SGS speaker
model_info = loader.load_models(model="sgs_10")
print(f"Training data: {model_info['model_info']['training_data']}")