/whisper_service

whisper封装的flask服务

Primary LanguagePython

whisper 服务

基于 whisper 封装函数提供 API 服务

环境要求

  1. python 3.7+

运行

docker pull wuhanchu/whisper_server
docker run -p 5000:5000 -d --name whisper_server wuhanchu/whisper_server

volume

//.cache/whisper 模型目录 /var/whipser 临时文件夹

环境变量

分组 配置项 说明
模型参数 MODEL 需要调用的模型名称 默认 base
文件参数 TEMP_FOLDER 临时文件夹

文件目录说明

├── README.md -- 项目说明
├── run.py -- 程序运行文件
├── run.sh -- 容器运行脚本
├── requirements.txt -- 项目使用到的依赖包
├── config.py -- 项目配置文件
├── Dockerfile -- 项目镜像构建文件

编译

初始化编译

docker build  . -t wuhanchu/whisper_server

更新编译

docker build -f ./Dockerfile.continue . -t wuhanchu/whisper_server
docker push wuhanchu/whisper_server

cuda10.2

docker build -f ./Dockerfile_cuda10.2.continue . -t wuhanchu/whisper_server:cuda10.2
docker push wuhanchu/whisper_server:cuda10.2