70 lines
1.7 KiB
Bash
70 lines
1.7 KiB
Bash
#!/bin/bash
|
|
# FunASR 安装脚本
|
|
# 用途：创建虚拟环境并安装 FunASR 及其依赖
|
|
|
|
set -e
|
|
|
|
# 配置
|
|
VENV_DIR="$HOME/.openclaw/workspace/funasr_env"
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
echo "=========================================="
|
|
echo "FunASR 安装脚本"
|
|
echo "=========================================="
|
|
echo ""
|
|
|
|
# 检查 Python
|
|
if ! command -v python3 &> /dev/null; then
|
|
echo "❌ 错误：未找到 python3"
|
|
echo "请先安装 Python 3.7+"
|
|
exit 1
|
|
fi
|
|
|
|
PYTHON_VERSION=$(python3 -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')")
|
|
echo "✓ Python 版本: $PYTHON_VERSION"
|
|
|
|
# 创建虚拟环境
|
|
if [ -d "$VENV_DIR" ]; then
|
|
echo "⚠️ 虚拟环境已存在: $VENV_DIR"
|
|
read -p "是否重新安装？(y/N): " -n 1 -r
|
|
echo
|
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
|
echo "安装已取消"
|
|
exit 0
|
|
fi
|
|
rm -rf "$VENV_DIR"
|
|
fi
|
|
|
|
echo "创建虚拟环境: $VENV_DIR"
|
|
python3 -m venv "$VENV_DIR"
|
|
source "$VENV_DIR/bin/activate"
|
|
|
|
# 升级 pip
|
|
echo ""
|
|
echo "升级 pip..."
|
|
pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
# 安装依赖
|
|
echo ""
|
|
echo "安装 FunASR 及依赖（这需要几分钟）..."
|
|
pip install funasr modelscope huggingface_hub torch torchaudio \
|
|
-i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
# 验证安装
|
|
echo ""
|
|
echo "验证安装..."
|
|
python3 -c "from funasr import AutoModel; print('✓ FunASR 安装成功')" || {
|
|
echo "❌ FunASR 安装失败"
|
|
exit 1
|
|
}
|
|
|
|
# 完成
|
|
echo ""
|
|
echo "=========================================="
|
|
echo "✓ 安装完成！"
|
|
echo "=========================================="
|
|
echo ""
|
|
echo "现在可以使用转录功能："
|
|
echo " bash $SCRIPT_DIR/transcribe.sh /path/to/audio.ogg"
|
|
echo ""
|