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