TaiYouWeb commited on
Commit
e2023c8
·
verified ·
1 Parent(s): 95f671a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -26,7 +26,7 @@ model = AutoModel(
26
  trust_remote_code=False,
27
  remote_code="./model.py",
28
  vad_model="fsmn-vad",
29
- punc_model="ct-punc",
30
  vad_kwargs={"max_single_segment_time": 15000},
31
  ncpu=torch.get_num_threads(),
32
  batch_size=1,
@@ -34,7 +34,7 @@ model = AutoModel(
34
  device=device,
35
  )
36
 
37
- def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", vad_kwargs='{"max_single_segment_time": 15000}',
38
  batch_size=1, language="auto", use_itn=True, batch_size_s=60,
39
  merge_vad=True, merge_length_s=15, batch_size_threshold_s=50,
40
  hotword=" ", ban_emo_unk=True):
@@ -66,7 +66,7 @@ def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", vad_
66
  inputs = [
67
  gr.Audio(type="filepath"),
68
  gr.Textbox(value="fsmn-vad", label="VAD Model"),
69
- gr.Textbox(value="ct-punc", label="PUNC Model"),
70
  gr.Textbox(value='{"max_single_segment_time": 15000}', label="VAD Kwargs"),
71
  gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
72
  gr.Textbox(value="auto", label="Language"),
 
26
  trust_remote_code=False,
27
  remote_code="./model.py",
28
  vad_model="fsmn-vad",
29
+ punc_model="iic/speech_campplus_sv_zh-cn_16k-common",
30
  vad_kwargs={"max_single_segment_time": 15000},
31
  ncpu=torch.get_num_threads(),
32
  batch_size=1,
 
34
  device=device,
35
  )
36
 
37
+ def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="iic/speech_campplus_sv_zh-cn_16k-common", vad_kwargs='{"max_single_segment_time": 15000}',
38
  batch_size=1, language="auto", use_itn=True, batch_size_s=60,
39
  merge_vad=True, merge_length_s=15, batch_size_threshold_s=50,
40
  hotword=" ", ban_emo_unk=True):
 
66
  inputs = [
67
  gr.Audio(type="filepath"),
68
  gr.Textbox(value="fsmn-vad", label="VAD Model"),
69
+ gr.Textbox(value="iic/speech_campplus_sv_zh-cn_16k-common", label="PUNC Model"),
70
  gr.Textbox(value='{"max_single_segment_time": 15000}', label="VAD Kwargs"),
71
  gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
72
  gr.Textbox(value="auto", label="Language"),