File size: 590 Bytes
db0d138 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from abc import ABC, abstractmethod
from enum import Enum
class AbstractASRModel(ABC):
def __init__(self, device='cpu'):
self.device = device
self.name = "AbstractASRModel"
@abstractmethod
def load(self, model_dir, language):
raise NotImplementedError
@abstractmethod
def transcribe(self, wav, language):
raise NotImplementedError
class ModelName(Enum):
WHISPER = "whisper"
WHISPER_FINETUNED = "whisper_finetuned"
FUNASR_NANO = "funasr_nano"
FUNASR_MLT_NANO = "funasr_mlt_nano"
FUNASR_QUANT = "funasr_quant"
|