Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- .dockerignore +6 -10
- .gitignore +4 -4
- src/vocalizr/__init__.py +18 -2
.dockerignore
CHANGED
@@ -1,13 +1,9 @@
|
|
1 |
-
.flox/
|
2 |
-
.github/
|
3 |
-
.vscode/
|
4 |
-
renovate.json
|
5 |
-
tmp/
|
6 |
-
.ruff_cache/
|
7 |
-
.mypy_cache/
|
8 |
**/__pycache__/
|
9 |
-
|
|
|
10 |
.idea/
|
|
|
|
|
11 |
.venv/
|
12 |
-
.
|
13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
**/__pycache__/
|
2 |
+
.deepsource.toml
|
3 |
+
.github/
|
4 |
.idea/
|
5 |
+
.mypy_cache/
|
6 |
+
.ruff_cache/
|
7 |
.venv/
|
8 |
+
.vscode/
|
9 |
+
renovate.json
|
.gitignore
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
-
tmp/
|
2 |
-
.venv/
|
3 |
-
.ruff_cache/
|
4 |
-
.mypy_cache/
|
5 |
**/__pycache__/
|
6 |
.env
|
|
|
|
|
|
|
|
|
7 |
results/.github
|
8 |
.idea
|
9 |
docker-bake.hcl
|
|
|
|
|
|
|
|
|
|
|
1 |
**/__pycache__/
|
2 |
.env
|
3 |
+
.mypy_cache/
|
4 |
+
.ruff_cache/
|
5 |
+
.venv/
|
6 |
+
logs/
|
7 |
results/.github
|
8 |
.idea
|
9 |
docker-bake.hcl
|
src/vocalizr/__init__.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from os import getenv
|
2 |
from pathlib import Path
|
3 |
|
@@ -8,15 +9,30 @@ from torch import cuda
|
|
8 |
|
9 |
load_dotenv()
|
10 |
|
11 |
-
BASE_DIR: Path = Path(__file__).parent.parent.parent
|
12 |
DEBUG: bool = getenv(key="DEBUG", default="False").lower() == "true"
|
13 |
CHAR_LIMIT: int = int(getenv(key="CHAR_LIMIT", default="-1"))
|
14 |
SERVER_NAME: str = getenv(key="GRADIO_SERVER_NAME", default="localhost")
|
15 |
SERVER_PORT: int = int(getenv(key="GRADIO_SERVER_PORT", default="8080"))
|
16 |
PIPELINE: KPipeline = KPipeline(lang_code="a", repo_id="hexgrad/Kokoro-82M")
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
|
|
|
|
|
19 |
logger.info(f"CUDA Available: {CUDA_AVAILABLE}")
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
CHOICES: dict[str, str] = {
|
22 |
"🇺🇸 🚺 Heart ❤️": "af_heart",
|
|
|
1 |
+
from datetime import datetime
|
2 |
from os import getenv
|
3 |
from pathlib import Path
|
4 |
|
|
|
9 |
|
10 |
load_dotenv()
|
11 |
|
|
|
12 |
DEBUG: bool = getenv(key="DEBUG", default="False").lower() == "true"
|
13 |
CHAR_LIMIT: int = int(getenv(key="CHAR_LIMIT", default="-1"))
|
14 |
SERVER_NAME: str = getenv(key="GRADIO_SERVER_NAME", default="localhost")
|
15 |
SERVER_PORT: int = int(getenv(key="GRADIO_SERVER_PORT", default="8080"))
|
16 |
PIPELINE: KPipeline = KPipeline(lang_code="a", repo_id="hexgrad/Kokoro-82M")
|
17 |
+
CURRENT_DATE: str = datetime.now().strftime(format="%Y-%m-%d_%H-%M-%S")
|
18 |
+
|
19 |
+
BASE_DIR: Path = Path(__file__).parent.parent.parent
|
20 |
+
RESULTS_DIR: Path = BASE_DIR / "results"
|
21 |
+
LOG_DIR: Path = BASE_DIR / "logs"
|
22 |
+
AUDIO_FILE_PATH: Path = RESULTS_DIR / f"{CURRENT_DATE}.wav"
|
23 |
+
LOG_FILE_PATH: Path = LOG_DIR / f"{CURRENT_DATE}.log"
|
24 |
+
|
25 |
+
RESULTS_DIR.mkdir(exist_ok=True)
|
26 |
+
LOG_DIR.mkdir(exist_ok=True)
|
27 |
|
28 |
+
CUDA_AVAILABLE: bool = cuda.is_available()
|
29 |
+
logger.add(LOG_FILE_PATH, format="{time:YYYY-MM-DD at HH:mm:ss} | {level} | {message}",colorize=True)
|
30 |
logger.info(f"CUDA Available: {CUDA_AVAILABLE}")
|
31 |
+
logger.info(f"Char limit: {CHAR_LIMIT}")
|
32 |
+
logger.info(f"Base directory: {BASE_DIR}")
|
33 |
+
logger.info(f"Results directory: {RESULTS_DIR}")
|
34 |
+
logger.info(f"Current date: {CURRENT_DATE}")
|
35 |
+
logger.info(f"Audio file path: {AUDIO_FILE_PATH}")
|
36 |
|
37 |
CHOICES: dict[str, str] = {
|
38 |
"🇺🇸 🚺 Heart ❤️": "af_heart",
|