Hubert Base Superb Ks

🚀 基于ONNX权重适配Transformers.js的Hubert模型

本项目将 superb/hubert-base-superb-ks 模型转换为ONNX权重，以适配 Transformers.js 库，方便在Web环境中使用。

🚀 快速开始

安装依赖

如果你还没有安装 Transformers.js JavaScript 库，可以使用以下命令从 NPM 进行安装：

npm i @huggingface/transformers

💻 使用示例

基础用法

以下是使用 Xenova/hubert-base-superb-ks 模型进行语音命令识别的示例代码：

import { pipeline } from '@huggingface/transformers';

// 创建音频分类管道
const classifier = await pipeline('audio-classification', 'Xenova/hubert-base-superb-ks');

// 对音频进行分类
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/speech-commands_down.wav';
const output = await classifier(url, { top_k: 5 });
// [
//   { label: 'down', score: 0.9954305291175842 },
//   { label: 'go', score: 0.004518700763583183 },
//   { label: '_unknown_', score: 0.00005029444946558215 },
//   { label: 'no', score: 4.877569494965428e-7 },
//   { label: 'stop', score: 5.504634081887616e-9 }
// ]

📚 详细文档

注意事项

⚠️ 重要提示

目前为ONNX权重单独创建一个仓库是临时解决方案，直到WebML获得更广泛的应用。如果你想让你的模型适用于Web环境，我们建议使用 🤗 Optimum 将模型转换为ONNX格式，并按照本仓库的结构进行组织（将ONNX权重放在名为 onnx 的子文件夹中）。

📦 模型信息