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 的子文件夾中）。

📦 模型信息