Spaces:
Running
on
Zero
Running
on
Zero
indiejoseph
commited on
Commit
•
0e97ff8
1
Parent(s):
972e738
Update transcriber/AutoTranscriber.py
Browse files
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:
|