| import logging | |
| from pydantic import BaseModel | |
| from pydantic_settings import BaseSettings | |
| from typing import Optional, Literal | |
| logger = logging.getLogger(__name__) | |
| class ModelSettings(BaseSettings): | |
| asr_model: str = "distil-whisper/distil-large-v3" | |
| class InferenceConfig(BaseModel): | |
| task: Literal["transcribe", "translate"] = "transcribe" | |
| batch_size: int = 24 | |
| chunk_length_s: int = 30 | |
| sampling_rate: int = 16000 | |
| language: Optional[str] = None | |
| model_settings = ModelSettings() | |
| logger.info(f"asr model: {model_settings.asr_model}") |