HG-ha / SenseVoice-Api

阿里SenseVoice的fastpi封装,采用onnx发布,附带量化模型,支持GPU。支持从URL文件进行语音识别。
46 stars 4 forks source link

SenseVoice-Api

此项目是基于SenseVoice的funasr_onnx版本进行的api发布, 使用Python3.10.14开发,可以快速发布到各种onnx支持的设备平台

下载模型文件:https://huggingface.co/mingl/Sensevoice_Api

目录结构:

SenseVoice-Api/
  │——Dockerfile
  │——main.py
  │——requirements.txt
  │——iic
      └─SenseVoiceSmall
            │——am.mvn
            │——chn_jpn_yue_eng_ko_spectok.bpe.model
            │——config.yaml
            │——model.onnx
            │——model_quant.onnx

SenseVoice

SenseVoice是具有音频理解能力的音频基础模型,包括语音识别(ASR)、语种识别(LID)、语音情感识别(SER)和声学事件分类(AEC)或声学事件检测(AED)。本项目提供SenseVoice模型的介绍以及在多个任务测试集上的benchmark,以及体验模型所需的环境安装的与推理方式。

核心功能 🎯

SenseVoice专注于高精度多语言语音识别、情感辨识和音频事件检测

Docker部署(CPU+量化模型)

# 国内
docker pull registry.cn-hangzhou.aliyuncs.com/yiminger/sensevoice:latest
docker run -p 8000:8000 registry.cn-hangzhou.aliyuncs.com/yiminger/sensevoice:latest

# Docker hub
docker pull yiminger/sensevoice:latest
# 运行
docker run -p 8000:8000 yiminger/sensevoice:latest

本地安装

git clone https://github.com/HG-ha/SenseVoice-Api.git && cd SenseVoice-Api
# 安装依赖
pip install -r requirements.txt
# 运行
python main.py

接口测试

  1. 从URL转文字
    curl --location --request POST 'http://127.0.0.1:8000/extract_text' \
    --form 'url=https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav'
  2. 从文件转文字
    curl --request POST \
    --url http://127.0.0.1:8000/extract_text \
    --header 'content-type: multipart/form-data' \
    --form 'file=@asr_example_zh.wav'

接口文档

http://127.0.0.1:8000/docs