afurkank commited on
Commit
d473317
1 Parent(s): 304c741

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +3 -13
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(asr_model = "distil-whisper/distil-large-v3")
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}")