Update preload_model.py

preload_model.py

@@ -8,32 +8,27 @@ import torch
 from kokoro import KModel, KPipeline
 import logging
 
+# SET CACHE ENVIRONMENT VARIABLES AT MODULE LEVEL - BEFORE ANY IMPORTS
+CACHE_DIR = "/app/cache"
+os.makedirs(CACHE_DIR, exist_ok=True)
+os.environ['HF_HOME'] = CACHE_DIR
+os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 def preload_model():
     """Pre-load the model to simple cache directory"""
-    cache_dir = "/app/cache"
-    os.makedirs(cache_dir, exist_ok=True)
-    
-    # CRITICAL: Set cache environment variables FIRST
-    os.environ['HF_HOME'] = cache_dir
-    os.environ['HUGGINGFACE_HUB_CACHE'] = cache_dir
-    os.environ['HF_HUB_CACHE'] = cache_dir
-    
-    logger.info(f"Pre-loading Kokoro-82M model to {cache_dir}...")
+    logger.info(f"Pre-loading Kokoro-82M model to {CACHE_DIR}...")
     
     try:
-        # Initialize model with explicit cache directory
+        # Initialize model (no cache_dir parameter needed)
         logger.info("Downloading model...")
-        model = KModel(
-            repo_id='hexgrad/Kokoro-82M',
-            cache_dir=cache_dir
-        ).to('cpu').eval()
+        model = KModel(repo_id='hexgrad/Kokoro-82M').to('cpu').eval()
         
         # Initialize pipeline
         logger.info("Initializing pipeline...")
-        pipeline = KPipeline(lang_code='a', model=False, cache_dir=cache_dir)
+        pipeline = KPipeline(lang_code='a', model=False)
         
         # Test with a small text
         test_text = "Hello world"