23 lines
725 B
Python
23 lines
725 B
Python
|
from ahserver.serverenv import ServerEnv
|
||
|
|
from appPublic.worker import awaitify
|
||
|
|
from appPublic.jsonConfig import getConfig, get_definition
|
||
|
|
from funasr import AutoModel
|
||
|
|
from funasr.utils.postprocess_utils import rich_transcription_postprocess
|
||
|
|
|
||
|
|
class SenseVoiceBase:
|
||
|
|
def __init__(self):
|
||
|
|
model_dir = get_definition('sensevoice_path')
|
||
|
|
self.model = AutoModel(model=model_dir)
|
||
|
|
|
||
|
|
def _stt(self, x):
|
||
|
|
pass
|
||
|
|
stt = awaitify(_stt)
|
||
|
|
|
||
|
|
class SenseVoiceFile(SenseVoiceBase):
|
||
|
|
def _stt(self, filepath):
|
||
|
|
res = this.model.generate(input=filepath,language="auto",
|
||
|
|
use_itn=True,
|
||
|
|
batch_size_s=60
|
||
|
|
)
|
||
|
|
return rich_transcription_postprocess(res[0]["text"])
|