Add Swift API for speech enhancement GTCRN models (#1989)

This commit is contained in:
Fangjun Kuang
2025-03-11 18:03:13 +08:00
committed by GitHub
parent 802119db17
commit c12d1d88c0
5 changed files with 192 additions and 0 deletions

View File

@@ -0,0 +1,36 @@
#!/usr/bin/env bash
set -ex
if [ ! -d ../build-swift-macos ]; then
echo "Please run ../build-swift-macos.sh first!"
exit 1
fi
if [ ! -f ./gtcrn_simple.onnx ]; then
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/speech-enhancement-models/gtcrn_simple.onnx
fi
if [ ! -f ./inp_16k.wav ]; then
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/speech-enhancement-models/inp_16k.wav
fi
if [ ! -e ./speech-enhancement-gtcrn ]; then
# Note: We use -lc++ to link against libc++ instead of libstdc++
swiftc \
-lc++ \
-I ../build-swift-macos/install/include \
-import-objc-header ./SherpaOnnx-Bridging-Header.h \
./speech-enhancement-gtcrn.swift ./SherpaOnnx.swift \
-L ../build-swift-macos/install/lib/ \
-l sherpa-onnx \
-l onnxruntime \
-o speech-enhancement-gtcrn
strip speech-enhancement-gtcrn
else
echo "./speech-enhancement-gtcrn exists - skip building"
fi
export DYLD_LIBRARY_PATH=$PWD/../build-swift-macos/install/lib:$DYLD_LIBRARY_PATH
./speech-enhancement-gtcrn