|
|
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" |
|
|
|
