MH0386 commited on
Commit
4008208
·
verified ·
1 Parent(s): 0a1c297

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. .dockerignore +6 -10
  2. .gitignore +4 -4
  3. 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
- dist/
 
10
  .idea/
 
 
11
  .venv/
12
- .deepsource.toml
13
- docker-bake.hcl
 
 
 
 
 
 
 
 
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
- CUDA_AVAILABLE: bool = cuda.is_available()
 
 
 
 
 
 
 
 
 
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",