Spaces:
Running on Zero
Running on Zero
| """Tải model/thư viện cho các VAD neural vào apps/gradio-lab/models (gitignore). | |
| Cùng phiên bản với packages/vad/scripts/fetch-models.sh của server Node, kiểm tra SHA-256. | |
| TEN VAD dùng thư viện native (ctypes) thay vì bản WebAssembly: chỉ có Linux x64 và macOS. | |
| uv run python scripts/fetch_models.py | |
| """ | |
| import hashlib | |
| import platform | |
| import sys | |
| import urllib.request | |
| from pathlib import Path | |
| MODELS_DIR = Path(__file__).resolve().parent.parent / "models" | |
| SILERO_VERSION = "v6.2.3" | |
| TEN_VAD_COMMIT = "22a3bcd4509d0faaa8eef4881e8af5f39c178950" | |
| TEN = f"https://raw.githubusercontent.com/TEN-framework/ten-vad/{TEN_VAD_COMMIT}" | |
| TEN_LIBS = { | |
| "Linux": ( | |
| "lib/Linux/x64/libten_vad.so", | |
| "5abfe6bf6e9a4fcea6b440240f0a9a0f431ab5006e48a4e16465ebe681ffd90f", | |
| ), | |
| "Darwin": ( | |
| "lib/macOS/ten_vad.framework/Versions/A/ten_vad", | |
| "81b2de13710670bb94fef315ab50fedc903a21c04c4290c6c2ac28d8b42e715a", | |
| ), | |
| } | |
| def fetch(url: str, dest: Path, sha256: str | None) -> None: | |
| if dest.exists() and (sha256 is None or _sha256(dest) == sha256): | |
| print(f"✓ {dest.relative_to(MODELS_DIR)} (đã có)") | |
| return | |
| dest.parent.mkdir(parents=True, exist_ok=True) | |
| tmp = dest.with_suffix(dest.suffix + ".tmp") | |
| urllib.request.urlretrieve(url, tmp) | |
| if sha256 is not None and _sha256(tmp) != sha256: | |
| tmp.unlink() | |
| sys.exit(f"✗ {dest.name}: sai checksum — file upstream đã đổi?") | |
| tmp.rename(dest) | |
| print(f"✓ {dest.relative_to(MODELS_DIR)}") | |
| def _sha256(path: Path) -> str: | |
| return hashlib.sha256(path.read_bytes()).hexdigest() | |
| def main() -> None: | |
| # Silero VAD — MIT license | |
| fetch( | |
| f"https://raw.githubusercontent.com/snakers4/silero-vad/{SILERO_VERSION}/src/silero_vad/data/silero_vad.onnx", | |
| MODELS_DIR / "silero_vad.onnx", | |
| "1a153a22f4509e292a94e67d6f9b85e8deb25b4988682b7e174c65279d8788e3", | |
| ) | |
| # TEN VAD — Apache 2.0 KÈM ĐIỀU KIỆN của Agora, xem models/ten-vad/LICENSE | |
| system = platform.system() | |
| if system not in TEN_LIBS or (system == "Linux" and platform.machine() not in ("x86_64", "AMD64")): | |
| print(f"! TEN VAD không có bản native cho {system}/{platform.machine()} — bỏ qua") | |
| return | |
| path, sha = TEN_LIBS[system] | |
| fetch(f"{TEN}/{path}", MODELS_DIR / "ten-vad" / Path(path).name, sha) | |
| fetch(f"{TEN}/LICENSE", MODELS_DIR / "ten-vad" / "LICENSE", None) | |
| if __name__ == "__main__": | |
| main() | |