Upload folder using huggingface_hub

pyproject.toml

@@ -28,6 +28,7 @@ dev = [
     "pylint>=3.3.7",
     "pyright>=1.1.400",
     "huggingface-hub[cli,hf-transfer]>=0.31.1",
+    "isort>=6.0.1",
 ]
 
 [tool.typos.default.extend-words]

src/vocalizr/__init__.py

@@ -4,10 +4,8 @@ from pathlib import Path
 from dotenv import load_dotenv
 from kokoro import KPipeline
 from loguru import logger
-from spacy import load
 from torch import cuda
 
-load(name="en_core_web_sm")
 load_dotenv()
 
 BASE_DIR: Path = Path(__file__).parent.parent.parent
@@ -50,4 +48,3 @@ CHOICES: dict[str, str] = {
     "🇬🇧 🚹 Lewis": "bm_lewis",
     "🇬🇧 🚹 Daniel": "bm_daniel",
 }
-}

src/vocalizr/model.py

@@ -21,16 +21,18 @@ def save_file_wav(audio: ndarray) -> None:
     :raise OSError: If an error occurs while saving the file.
     """
     makedirs(name="results", exist_ok=True)
-    filename = f"{BASE_DIR}/results/{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.wav"
+    filename: str = (
+        f"{BASE_DIR}/results/{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.wav"
+    )
     try:
         logger.info(f"Saving audio to {filename}")
-        write(filename, audio, 24000)
+        write(file=filename, data=audio, samplerate=24000)
     except OSError as e:
         raise OSError(f"Failed to save audio to {filename}: {e}") from e
 
 
 def generate_audio_for_text(
-    text: str, voice="af_heart", speed=1, save_file: bool = False
+    text: str, voice: str = "af_heart", speed: float = 1, save_file: bool = False
 ) -> tuple[int, ndarray]:
     """Generate audio for the input text.
 
@@ -46,8 +48,8 @@ def generate_audio_for_text(
         for _, _, audio in PIPELINE(text, voice, speed):
             audio = Tensor(audio).numpy()
             if save_file:
-                save_file_wav(audio)
+                save_file_wav(audio=audio)
             return 24000, audio
     except Error as e:
-        raise Error(str(e)) from e
+        raise Error(message=str(e)) from e
     raise RuntimeError("No audio generated")

uv.lock

@@ -2302,6 +2302,7 @@ dependencies = [
 dev = [
     { name = "black" },
     { name = "huggingface-hub", extra = ["cli", "hf-transfer"] },
+    { name = "isort" },
     { name = "mypy" },
     { name = "pylint" },
     { name = "pyrefly" },
@@ -2322,6 +2323,7 @@ requires-dist = [
 dev = [
     { name = "black", specifier = ">=25.1.0" },
     { name = "huggingface-hub", extras = ["cli", "hf-transfer"], specifier = ">=0.31.1" },
+    { name = "isort", specifier = ">=6.0.1" },
     { name = "mypy", specifier = ">=1.15.0" },
     { name = "pylint", specifier = ">=3.3.7" },
     { name = "pyrefly", specifier = ">=0.14.0" },