File size: 1,036 Bytes
5efbc82
 
 
 
 
 
 
 
 
 
 
 
 
1b567fa
 
 
 
 
 
5efbc82
 
 
1b567fa
5efbc82
 
 
1b567fa
5efbc82
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
"""
Configuration for Kokoro TTS API, especially for Hugging Face Spaces deployment.
"""

import os
import tempfile
import logging

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def setup_hf_cache():
    """Setup cache environment variables for Hugging Face Spaces"""
    # Don't try to create custom directories, just set environment variables
    # HF Spaces will handle the actual cache locations
    cache_settings = {
        'NUMBA_DISABLE_JIT': '1',
        'HF_HUB_DISABLE_TELEMETRY': '1'
    }
    
    # Set environment variables
    for key, value in cache_settings.items():
        os.environ[key] = value
        logger.info(f"Set {key} to {value}")
    
    logger.info("Cache environment setup completed")

def get_temp_dir():
    """Get a writable temporary directory"""
    return tempfile.gettempdir()

def is_hf_spaces():
    """Check if running on Hugging Face Spaces"""
    return os.environ.get('SPACE_ID') is not None

# Initialize cache setup
setup_hf_cache()