Wav2vec2 Base Superb Ks

🚀 基於ONNX權重適配Transformers.js的音頻模型

本項目將 superb/wav2vec2-base-superb-ks 模型轉換為ONNX權重，以適配 Transformers.js 庫，方便在Web環境中使用該音頻模型進行相關任務。

🚀 快速開始

本項目使用 Transformers.js 庫實現音頻分類功能，以下是快速開始的步驟。

📦 安裝指南

如果你還未安裝 Transformers.js JavaScript庫，可以使用以下命令從 NPM 進行安裝：

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 },
// ]

📚 詳細文檔

注意事項

目前為ONNX權重單獨創建一個倉庫是一種臨時解決方案，直到WebML獲得更廣泛的應用。如果你想讓你的模型適用於Web環境，我們建議使用 🤗 Optimum 將模型轉換為ONNX格式，並按照本倉庫的結構進行組織（將ONNX權重放在名為 onnx 的子文件夾中）。

模型信息