Support paraformer. (#95)

.github/scripts/test-offline-transducer.sh

@@ -25,36 +25,59 @@ log "Download pretrained model and test-data from $repo_url"
 GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url
 pushd $repo
 git lfs pull --include "*.onnx"
-cd test_wavs
 popd
 
-waves=(
-$repo/test_wavs/0.wav
-$repo/test_wavs/1.wav
-$repo/test_wavs/2.wav
-)
-
-for wave in ${waves[@]}; do
-  time $EXE \
-  $repo/tokens.txt \
-  $repo/encoder-epoch-99-avg-1.onnx \
-  $repo/decoder-epoch-99-avg-1.onnx \
-  $repo/joiner-epoch-99-avg-1.onnx \
-  $wave \
-  2
-done
+time $EXE \
+  --tokens=$repo/tokens.txt \
+  --encoder=$repo/encoder-epoch-99-avg-1.onnx \
+  --decoder=$repo/decoder-epoch-99-avg-1.onnx \
+  --joiner=$repo/joiner-epoch-99-avg-1.onnx \
+  --num-threads=2 \
+  $repo/test_wavs/0.wav \
+  $repo/test_wavs/1.wav \
+  $repo/test_wavs/2.wav
 
 
 if command -v sox &> /dev/null; then
   echo "test 8kHz"
   sox $repo/test_wavs/0.wav -r 8000 8k.wav
+
   time $EXE \
-    $repo/tokens.txt \
-    $repo/encoder-epoch-99-avg-1.onnx \
-    $repo/decoder-epoch-99-avg-1.onnx \
-    $repo/joiner-epoch-99-avg-1.onnx \
-    8k.wav \
-    2
+    --tokens=$repo/tokens.txt \
+    --encoder=$repo/encoder-epoch-99-avg-1.onnx \
+    --decoder=$repo/decoder-epoch-99-avg-1.onnx \
+    --joiner=$repo/joiner-epoch-99-avg-1.onnx \
+    --num-threads=2 \
+    $repo/test_wavs/0.wav \
+    $repo/test_wavs/1.wav \
+    $repo/test_wavs/2.wav \
+    8k.wav
 fi
 
 rm -rf $repo
+
+log "------------------------------------------------------------"
+log "Run Paraformer (Chinese)"
+log "------------------------------------------------------------"
+
+repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28
+log "Start testing ${repo_url}"
+repo=$(basename $repo_url)
+log "Download pretrained model and test-data from $repo_url"
+
+GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url
+pushd $repo
+git lfs pull --include "*.onnx"
+popd
+
+time $EXE \
+  --tokens=$repo/tokens.txt \
+  --paraformer=$repo/model.onnx \
+  --num-threads=2 \
+  --decoding-method=greedy_search \
+  $repo/test_wavs/0.wav \
+  $repo/test_wavs/1.wav \
+  $repo/test_wavs/2.wav \
+  $repo/test_wavs/8k.wav
+
+rm -rf $repo

.github/workflows/windows-x64.yaml

@@ -71,7 +71,15 @@ jobs:
 
           ls -lh ./bin/Release/sherpa-onnx.exe
 
-      - name: Test sherpa-onnx for Windows x64
+      - name: Test offline transducer for Windows x64
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin/Release:$PATH
+          export EXE=sherpa-onnx-offline.exe
+
+          .github/scripts/test-offline-transducer.sh
+
+      - name: Test online transducer for Windows x64
         shell: bash
         run: |
           export PATH=$PWD/build/bin/Release:$PATH

.github/workflows/windows-x86.yaml

@@ -71,7 +71,15 @@ jobs:
 
           ls -lh ./bin/Release/sherpa-onnx.exe
 
-      - name: Test sherpa-onnx for Windows x86
+      - name: Test offline transducer for Windows x86
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin/Release:$PATH
+          export EXE=sherpa-onnx-offline.exe
+
+          .github/scripts/test-offline-transducer.sh
+
+      - name: Test online transducer for Windows x86
         shell: bash
         run: |
           export PATH=$PWD/build/bin/Release:$PATH