Hubert Base Superb Ks

🚀 ハバートベースの音声認識モデル（Transformers.js対応）

このプロジェクトは、superb/hubert-base-superb-ksモデルをTransformers.jsと互換性のあるONNXウェイトに変換したものです。Transformers.jsを使って音声コマンド認識などのタスクを実行できます。

🚀 クイックスタート

📦 インストール

まだインストールしていない場合は、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というサブフォルダに配置）。