#!/usr/bin/env python3
import os
import re
from pathlib import Path
from typing import List
BASE_URL = "https://huggingface.co/csukuangfj/sherpa-onnx-apk/resolve/main/"
from dataclasses import dataclass
@dataclass
class APK:
major: int
minor: int
patch: int
arch: str
short_name: str
def __init__(self, s):
# sherpa-onnx-1.9.23-arm64-v8a-silero_vad.apk
# sherpa-onnx-1.9.23-x86-silero_vad.apk
s = str(s)[len("asr/") :]
split = s.split("-")
self.major, self.minor, self.patch = list(map(int, split[2].split(".")))
self.arch = split[3]
self.short_name = split[4]
if "arm" in s:
self.arch += "-" + split[4]
self.short_name = split[5]
if "armeabi" in self.arch:
self.arch = "y" + self.arch
if "arm64" in self.arch:
self.arch = "z" + self.arch
def sort_by_apk(x):
x = APK(x)
return (x.major, x.minor, x.patch, x.arch, x.short_name)
def generate_url(files: List[str]) -> List[str]:
ans = []
base = BASE_URL
for f in files:
ans.append(base + str(f))
return ans
def get_all_files(d: str, suffix: str) -> List[str]:
ans = sorted(Path(d).glob(suffix), key=sort_by_apk, reverse=True)
return list(map(lambda x: BASE_URL + str(x), ans))
def to_file(filename: str, files: List[str]):
content = r"""
APKs for VAD
This page lists the VAD APKs for sherpa-onnx,
one of the deployment frameworks of the Next-gen Kaldi project.
The name of an APK has the following rule:
sherpa-onnx-{version}-{arch}-{model}.apk
where
version: It specifies the current version, e.g., 1.9.23
arch: The architecture targeted by this APK, e.g., arm64-v8a, armeabi-v7a, x86_64, x86
"""
with open(filename, "w") as f:
print(content, file=f)
for x in files:
name = x.rsplit("/", maxsplit=1)[-1]
print(f'{name} ', file=f)
def main():
apk = get_all_files("vad", suffix="*.apk")
to_file("./apk-vad.html", apk)
# for Chinese users
apk2 = []
for a in apk:
a = a.replace("huggingface.co", "hf-mirror.com")
a = a.replace("resolve", "blob")
apk2.append(a)
to_file("./apk-vad-cn.html", apk2)
if __name__ == "__main__":
main()