EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Flutter demo for real-time speech recognition (#1042)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-06-23 13:29:13 +08:00
committed by GitHub
parent 9dd0e03568
commit 169c9bf627
67 changed files with 4056 additions and 235 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
flutter-examples/streaming_asr/lib/online_model.dart Normal file
View File

@@ -0,0 +1,68 @@
import 'package:sherpa_onnx/sherpa_onnx.dart' as sherpa_onnx;
import './utils.dart';
// Remember to change `assets` in ../pubspec.yaml
// and download files to ../assets
Future<sherpa_onnx.OnlineModelConfig> getOnlineModelConfig(
{required int type}) async {
switch (type) {
case 0:
final modelDir =
'assets/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20';
return sherpa_onnx.OnlineModelConfig(
transducer: sherpa_onnx.OnlineTransducerModelConfig(
encoder:
await copyAssetFile('$modelDir/encoder-epoch-99-avg-1.int8.onnx'),
decoder: await copyAssetFile('$modelDir/decoder-epoch-99-avg-1.onnx'),
joiner: await copyAssetFile('$modelDir/joiner-epoch-99-avg-1.onnx'),
),
tokens: await copyAssetFile('$modelDir/tokens.txt'),
modelType: 'zipformer',
);
case 1:
final modelDir = 'assets/sherpa-onnx-streaming-zipformer-en-2023-06-26';
return sherpa_onnx.OnlineModelConfig(
transducer: sherpa_onnx.OnlineTransducerModelConfig(
encoder: await copyAssetFile(
'$modelDir/encoder-epoch-99-avg-1-chunk-16-left-128.int8.onnx'),
decoder: await copyAssetFile(
'$modelDir/decoder-epoch-99-avg-1-chunk-16-left-128.onnx'),
joiner: await copyAssetFile(
'$modelDir/joiner-epoch-99-avg-1-chunk-16-left-128.onnx'),
),
tokens: await copyAssetFile('$modelDir/tokens.txt'),
modelType: 'zipformer2',
);
case 2:
final modelDir =
'assets/icefall-asr-zipformer-streaming-wenetspeech-20230615';
return sherpa_onnx.OnlineModelConfig(
transducer: sherpa_onnx.OnlineTransducerModelConfig(
encoder: await copyAssetFile(
'$modelDir/exp/encoder-epoch-12-avg-4-chunk-16-left-128.int8.onnx'),
decoder: await copyAssetFile(
'$modelDir/exp/decoder-epoch-12-avg-4-chunk-16-left-128.onnx'),
joiner: await copyAssetFile(
'$modelDir/exp/joiner-epoch-12-avg-4-chunk-16-left-128.onnx'),
),
tokens: await copyAssetFile('$modelDir/data/lang_char/tokens.txt'),
modelType: 'zipformer2',
);
case 3:
final modelDir = 'assets/sherpa-onnx-streaming-zipformer-fr-2023-04-14';
return sherpa_onnx.OnlineModelConfig(
transducer: sherpa_onnx.OnlineTransducerModelConfig(
encoder: await copyAssetFile(
'$modelDir/encoder-epoch-29-avg-9-with-averaged-model.int8.onnx'),
decoder: await copyAssetFile(
'$modelDir/decoder-epoch-29-avg-9-with-averaged-model.onnx'),
joiner: await copyAssetFile(
'$modelDir/joincoder-epoch-29-avg-9-with-averaged-model.onnx'),
),
tokens: await copyAssetFile('$modelDir/tokens.txt'),
modelType: 'zipformer',
);
default:
throw ArgumentError('Unsupported type: $type');
}
}