Spaces:
Runtime error
Runtime error
File size: 2,723 Bytes
4008208 b9a47ba 7f44fc3 b9a47ba 2404bb8 b9a47ba 2404bb8 b9a47ba 967c1e0 4008208 1458ba2 72a7954 4008208 72a7954 4008208 b9a47ba 4008208 72a7954 b9a47ba 2404bb8 4008208 2404bb8 4008208 2404bb8 b9a47ba |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
from datetime import datetime
from os import getenv
from pathlib import Path
from warnings import filterwarnings
from dotenv import load_dotenv
from kokoro import KPipeline
from loguru import logger
from torch import cuda
filterwarnings(
action="ignore",
message="dropout option adds dropout after all but last recurrent layer",
)
filterwarnings(
action="ignore",
message="`torch.nn.utils.weight_norm` is deprecated",
)
load_dotenv()
DEBUG: bool = getenv(key="DEBUG", default="True").lower() == "true"
SERVER_NAME: str = getenv(key="GRADIO_SERVER_NAME", default="localhost")
SERVER_PORT: int = int(getenv(key="GRADIO_SERVER_PORT", default="8080"))
PIPELINE: KPipeline = KPipeline(lang_code="a", repo_id="hexgrad/Kokoro-82M")
CURRENT_DATE: str = datetime.now().strftime(format="%Y-%m-%d_%H-%M-%S")
BASE_DIR: Path = Path.cwd()
RESULTS_DIR: Path = BASE_DIR / "results"
LOG_DIR: Path = BASE_DIR / "logs"
AUDIO_FILE_PATH: Path = RESULTS_DIR / f"{CURRENT_DATE}.wav"
LOG_FILE_PATH: Path = LOG_DIR / f"{CURRENT_DATE}.log"
RESULTS_DIR.mkdir(exist_ok=True)
LOG_DIR.mkdir(exist_ok=True)
CUDA_AVAILABLE: bool = cuda.is_available()
logger.add(
LOG_FILE_PATH,
format="{time:YYYY-MM-DD at HH:mm:ss} | {level} | {message}",
colorize=True,
)
logger.info(f"CUDA Available: {CUDA_AVAILABLE}")
logger.info(f"Current date: {CURRENT_DATE}")
logger.info(f"Base directory: {BASE_DIR}")
logger.info(f"Results directory: {RESULTS_DIR}")
logger.info(f"Log directory: {LOG_DIR}")
logger.info(f"Audio file path: {AUDIO_FILE_PATH}")
logger.info(f"Log file path: {LOG_FILE_PATH}")
CHOICES: dict[str, str] = {
"🇺🇸 🚺 Heart ❤️": "af_heart",
"🇺🇸 🚺 Bella 🔥": "af_bella",
"🇺🇸 🚺 Nicole 🎧": "af_nicole",
"🇺🇸 🚺 Aoede": "af_aoede",
"🇺🇸 🚺 Kore": "af_kore",
"🇺🇸 🚺 Sarah": "af_sarah",
"🇺🇸 🚺 Nova": "af_nova",
"🇺🇸 🚺 Sky": "af_sky",
"🇺🇸 🚺 Alloy": "af_alloy",
"🇺🇸 🚺 Jessica": "af_jessica",
"🇺🇸 🚺 River": "af_river",
"🇺🇸 🚹 Michael": "am_michael",
"🇺🇸 🚹 Fenrir": "am_fenrir",
"🇺🇸 🚹 Puck": "am_puck",
"🇺🇸 🚹 Echo": "am_echo",
"🇺🇸 🚹 Eric": "am_eric",
"🇺🇸 🚹 Liam": "am_liam",
"🇺🇸 🚹 Onyx": "am_onyx",
"🇺🇸 🚹 Santa": "am_santa",
"🇺🇸 🚹 Adam": "am_adam",
"🇬🇧 🚺 Emma": "bf_emma",
"🇬🇧 🚺 Isabella": "bf_isabella",
"🇬🇧 🚺 Alice": "bf_alice",
"🇬🇧 🚺 Lily": "bf_lily",
"🇬🇧 🚹 George": "bm_george",
"🇬🇧 🚹 Fable": "bm_fable",
"🇬🇧 🚹 Lewis": "bm_lewis",
"🇬🇧 🚹 Daniel": "bm_daniel",
}
|