indiejoseph commited on
Commit
0e97ff8
1 Parent(s): 972e738

Update transcriber/AutoTranscriber.py

Browse files
Files changed (1) hide show
  1. transcriber/AutoTranscriber.py +2 -2
transcriber/AutoTranscriber.py CHANGED
@@ -40,7 +40,7 @@ class AutoTranscriber:
40
  self.asr_model = AutoModel(
41
  model="iic/SenseVoiceSmall",
42
  vad_model=None, # We'll handle VAD separately
43
- punc_model="ct-punc" if with_punct else None,
44
  ban_emo_unks=True,
45
  device=device,
46
  )
@@ -91,7 +91,7 @@ class AutoTranscriber:
91
 
92
  # Get ASR results for segment
93
  asr_result = self.asr_model.generate(
94
- input=segment_audio, language="yue", use_itn=True
95
  )
96
 
97
  if not asr_result:
 
40
  self.asr_model = AutoModel(
41
  model="iic/SenseVoiceSmall",
42
  vad_model=None, # We'll handle VAD separately
43
+ punc_model=None,
44
  ban_emo_unks=True,
45
  device=device,
46
  )
 
91
 
92
  # Get ASR results for segment
93
  asr_result = self.asr_model.generate(
94
+ input=segment_audio, language="yue", use_itn=self.with_punct
95
  )
96
 
97
  if not asr_result: