feat: asr blackblox

This commit is contained in:
superobk
2024-03-19 11:44:51 +08:00
parent 4e2a4ef63c
commit 2dccf5e78d
19 changed files with 1408 additions and 14 deletions

18
src/asr/asr_service.py Normal file
View File

@ -0,0 +1,18 @@
import io
import logging
from .rapid_paraformer import RapidParaformer
from .rapid_paraformer.utils import read_yaml
class ASRService():
def __init__(self, config_path: str):
config = read_yaml(config_path)
print(config)
logging.info('Initializing ASR Service...')
self.paraformer = RapidParaformer(config)
def infer(self, wav_path):
by = open(wav_path, 'rb')
result = self.paraformer([io.BytesIO(by.read())])
return result[0]