Spaces:
Sleeping
Sleeping
File size: 1,536 Bytes
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
"""
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 directories for Hugging Face Spaces"""
cache_dirs = {
'HF_HOME': '/tmp/hf_cache',
'TRANSFORMERS_CACHE': '/tmp/hf_cache',
'HF_HUB_CACHE': '/tmp/hf_cache',
'TORCH_HOME': '/tmp/torch_cache',
'NUMBA_CACHE_DIR': '/tmp/numba_cache'
}
# Set environment variables
for key, value in cache_dirs.items():
os.environ[key] = value
logger.info(f"Set {key} to {value}")
# Create directories
for cache_dir in set(cache_dirs.values()):
try:
os.makedirs(cache_dir, exist_ok=True)
# Ensure write permissions
os.chmod(cache_dir, 0o777)
logger.info(f"Created cache directory: {cache_dir}")
except Exception as e:
logger.warning(f"Could not create/modify {cache_dir}: {e}")
# Additional HF settings
os.environ['NUMBA_DISABLE_JIT'] = '1'
os.environ['HF_HUB_DISABLE_TELEMETRY'] = '1'
logger.info("Cache directories 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() |