Wav2vec2 Base Superb Ks

🚀 音声分類モデル（wav2vec2-base-superb-ks）

このプロジェクトは、superb/wav2vec2-base-superb-ksモデルをTransformers.jsと互換性のあるONNX形式に変換したものです。音声分類タスクに利用できます。

🚀 クイックスタート

📦 インストール

Transformers.jsを使うために、まずは以下のコマンドでJavaScriptライブラリをインストールします。

npm i @huggingface/transformers

💻 使用例

基本的な使用法

Xenova/wav2vec2-base-superb-ksを使って音声分類を行い、上位3つの結果を返す例です。

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

// 音声分類パイプラインを作成
const classifier = await pipeline('audio-classification', 'Xenova/wav2vec2-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: 3 });
console.log(output);
// [
//   { label: 'down', score: 0.9998697638511658 },
//   { label: 'go', score: 0.00009957332076737657 },
//   { label: '_unknown_', score: 0.000029320701287360862 },
// ]

注意事項

WebMLが普及するまで、ONNXウェイトを別のリポジトリにするのは一時的な解決策です。モデルをWeb対応にする場合は、🤗 Optimumを使ってONNX形式に変換し、このリポジトリのように構成することをおすすめします（ONNXウェイトはonnxというサブフォルダに配置）。