Update config.py
Browse files
config.py
CHANGED
@@ -8,26 +8,16 @@ logger = logging.getLogger(__name__)
|
|
8 |
|
9 |
|
10 |
class ModelSettings(BaseSettings):
|
11 |
-
asr_model: str
|
12 |
-
assistant_model: Optional[str]
|
13 |
-
diarization_model: Optional[str]
|
14 |
-
hf_token: Optional[str]
|
15 |
-
|
16 |
|
17 |
class InferenceConfig(BaseModel):
|
18 |
task: Literal["transcribe", "translate"] = "transcribe"
|
19 |
batch_size: int = 24
|
20 |
-
assisted: bool = False
|
21 |
chunk_length_s: int = 30
|
22 |
sampling_rate: int = 16000
|
23 |
language: Optional[str] = None
|
24 |
-
num_speakers: Optional[int] = None
|
25 |
-
min_speakers: Optional[int] = None
|
26 |
-
max_speakers: Optional[int] = None
|
27 |
|
28 |
|
29 |
-
model_settings = ModelSettings(
|
30 |
|
31 |
-
logger.info(f"asr model: {model_settings.asr_model}")
|
32 |
-
logger.info(f"assist model: {model_settings.assistant_model}")
|
33 |
-
logger.info(f"diar model: {model_settings.diarization_model}")
|
|
|
8 |
|
9 |
|
10 |
class ModelSettings(BaseSettings):
|
11 |
+
asr_model: str = "distil-whisper/distil-large-v3"
|
|
|
|
|
|
|
|
|
12 |
|
13 |
class InferenceConfig(BaseModel):
|
14 |
task: Literal["transcribe", "translate"] = "transcribe"
|
15 |
batch_size: int = 24
|
|
|
16 |
chunk_length_s: int = 30
|
17 |
sampling_rate: int = 16000
|
18 |
language: Optional[str] = None
|
|
|
|
|
|
|
19 |
|
20 |
|
21 |
+
model_settings = ModelSettings()
|
22 |
|
23 |
+
logger.info(f"asr model: {model_settings.asr_model}")
|
|
|
|