fix retract error (#1350)

This commit is contained in:
夜雨飘零 2024-02-03 13:05:07 +08:00 committed by GitHub
parent c2f174cd78
commit 2cf4084b23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,6 +4,7 @@ import os
import uuid
import aiofiles
import ffmpeg
import uvicorn
from fastapi import FastAPI, File, UploadFile
from modelscope.utils.logger import get_logger
@ -105,7 +106,6 @@ logger.info("loaded models!")
app = FastAPI(title="FunASR")
param_dict = {"sentence_timestamp": True, "batch_size_s": 300}
if args.hotword_path is not None and os.path.exists(args.hotword_path):
with open(args.hotword_path, 'r', encoding='utf-8') as f:
lines = f.readlines()
@ -122,8 +122,7 @@ async def api_recognition(audio: UploadFile = File(..., description="audio file"
async with aiofiles.open(audio_path, 'wb') as out_file:
content = await audio.read()
await out_file.write(content)
try:
try:
audio_bytes, _ = (
ffmpeg.input(audio_path, threads=0)
.output("-", format="s16le", acodec="pcm_s16le", ac=1, ar=16000)
@ -152,6 +151,5 @@ async def api_recognition(audio: UploadFile = File(..., description="audio file"
return {"msg": "未知错误", "code": -1}
if __name__ == '__main__':
uvicorn.run(app, host=args.host, port=args.port, ssl_keyfile=args.keyfile, ssl_certfile=args.certfile)