Update main.py

main.py

@@ -166,7 +166,35 @@ def convert_model_from_local(
     
     save_file(final_sd, save_path)
 
-
+from huggingface_hub import hf_hub_download
+import numpy as np
+import torch
+from pathlib import Path
+file_path = hf_hub_download(
+    repo_id="KittenML/kitten-tts-nano-0.8-fp32",
+    filename="voices.npz"
+)
+keys = [
+    'expr-voice-2-m', 'expr-voice-2-f',
+    'expr-voice-3-m', 'expr-voice-3-f',
+    'expr-voice-4-m', 'expr-voice-4-f',
+    'expr-voice-5-m', 'expr-voice-5-f'
+]
+
+values = [
+    'Bella', 'Jasper',
+    'Luna', 'Bruno',
+    'Rosie', 'Hugo',
+    'Kiki', 'Leo'
+]
+voice_dict = dict(zip(keys, values))
+voice_dir = Path("voice_dir")
+voice_dir.mkdir(exist_ok=True)
+data = np.load(file_path)
+for key in data.files:
+    if key in voice_dict:
+        tensor = torch.from_numpy(data[key])
+        torch.save(tensor, voice_dir / f"{voice_dict[key]}.pt")
 
 # config = StyleTTS2Config.from_yaml("config.yaml")
 # convert_model_from_local(