Spaces:
Running
Running
Update transcriber/AutoTranscriber.py
#1
by
indiejoseph - opened
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=
|
| 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=
|
| 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:
|