Update download_model.py

download_model.py

@@ -7,7 +7,7 @@ import platform
 
 # Repository ID
 repo_id = "hexgrad/Kokoro-82M"
-
+repo_id2="Remsky/kokoro-82m-mirror"
 # Set up the cache directory
 cache_dir = "./cache"
 os.makedirs(cache_dir, exist_ok=True)
@@ -37,8 +37,8 @@ def get_voice_models():
     """Downloads missing voice models from the Hugging Face repository."""
 
     # Create or empty the 'voices' directory
-    # if os.path.exists(VOICES_DIR):
-    #    shutil.rmtree(VOICES_DIR)
+    if os.path.exists(VOICES_DIR):
+       shutil.rmtree(VOICES_DIR)
     os.makedirs(VOICES_DIR, exist_ok=True)
 
     # Get list of files from the repository
@@ -55,14 +55,17 @@ def get_voice_models():
     if download_voice:
         #  print(f"Files to download: {download_voice}")
          pass
-    
-    download_files(repo_id, [f"voices/{file}" for file in download_voice], VOICES_DIR, cache_dir)
+    eng_voices = []
+    for i in download_voice:
+         if i.startswith("a") or i.startswith("b"):
+              eng_voices.append(i)        
+    download_files(repo_id, [f"voices/{file}" for file in eng_voices], VOICES_DIR, cache_dir)
 
 def download_base_models():
     """Downloads Kokoro base model and fp16 version if missing."""
 
-    download_files(repo_id, [KOKORO_FILE], KOKORO_DIR, cache_dir)
-    download_files(repo_id, [FP16_FILE], FP16_DIR, cache_dir)
+    download_files(repo_id2, [KOKORO_FILE], KOKORO_DIR, cache_dir)
+    download_files(repo_id2, [FP16_FILE], FP16_DIR, cache_dir)
 
 def setup_batch_file():
     """Creates a 'run_app.bat' file for Windows if it doesn't exist."""