AnyaSchen commited on
Commit
15421e7
·
1 Parent(s): 2cd6c79

fix: add args 2

Browse files
Files changed (1) hide show
  1. core.py +41 -24
core.py CHANGED
@@ -14,32 +14,49 @@ class WhisperLiveKit:
14
  if WhisperLiveKit._initialized:
15
  return
16
 
17
- if args is None:
18
- args = Namespace(
19
- host="localhost",
20
- port=8000,
21
- warmup_file=None,
22
- confidence_validation=False,
23
- diarization=False,
24
- transcription=True,
25
- min_chunk_size=0.5,
26
- model="base",
27
- model_cache_dir=None,
28
- model_dir=None,
29
- lan="en",
30
- task="transcribe",
31
- backend="faster-whisper",
32
- vac=False,
33
- vac_chunk_size=0.04,
34
- vad=True,
35
- buffer_trimming="sentence",
36
- buffer_trimming_sec=1.0,
37
- log_level="INFO"
38
- )
39
 
40
- self.args = args
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
41
 
42
- print(args)
43
 
44
  self.asr = None
45
  self.tokenizer = None
 
14
  if WhisperLiveKit._initialized:
15
  return
16
 
17
+ # Define default args
18
+ default_args = Namespace(
19
+ host="localhost",
20
+ port=8000,
21
+ warmup_file=None,
22
+ confidence_validation=False,
23
+ diarization=False,
24
+ transcription=True,
25
+ min_chunk_size=0.5,
26
+ model="base",
27
+ model_cache_dir=None,
28
+ model_dir=None,
29
+ lan="en",
30
+ task="transcribe",
31
+ backend="faster-whisper",
32
+ vac=False,
33
+ vac_chunk_size=0.04,
34
+ vad=True,
35
+ buffer_trimming="sentence",
36
+ buffer_trimming_sec=1.0,
37
+ log_level="INFO"
38
+ )
39
 
40
+ # If args is provided, merge it with defaults
41
+ if args is not None:
42
+ # Convert args to dict if it's a Namespace
43
+ if isinstance(args, Namespace):
44
+ args_dict = vars(args)
45
+ else:
46
+ args_dict = args
47
+
48
+ # Convert default_args to dict
49
+ default_dict = vars(default_args)
50
+
51
+ # Merge provided args with defaults
52
+ merged_dict = {**default_dict, **args_dict}
53
+
54
+ # Convert back to Namespace
55
+ self.args = Namespace(**merged_dict)
56
+ else:
57
+ self.args = default_args
58
 
59
+ print(self.args)
60
 
61
  self.asr = None
62
  self.tokenizer = None