TaiYouWeb commited on
Commit
a82fccb
1 Parent(s): 6b28214

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
30
  spk_model="cam++",
31
  vad_kwargs={"max_single_segment_time": 15000},
32
  ncpu=torch.get_num_threads(),
@@ -67,7 +67,7 @@ def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", spk_
67
  inputs = [
68
  gr.Audio(type="filepath"),
69
  gr.Textbox(value="fsmn-vad", label="VAD Model"),
70
- gr.Textbox(value="iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch", label="PUNC Model"),
71
  gr.Textbox(value="cam++", label="SPK Model"),
72
  gr.Textbox(value='{"max_single_segment_time": 15000}', label="VAD Kwargs"),
73
  gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
 
26
  trust_remote_code=False,
27
  remote_code="./model.py",
28
  vad_model="fsmn-vad",
29
+ punc_model="ct-punc",
30
  spk_model="cam++",
31
  vad_kwargs={"max_single_segment_time": 15000},
32
  ncpu=torch.get_num_threads(),
 
67
  inputs = [
68
  gr.Audio(type="filepath"),
69
  gr.Textbox(value="fsmn-vad", label="VAD Model"),
70
+ gr.Textbox(value="ct-punc", label="PUNC Model"),
71
  gr.Textbox(value="cam++", label="SPK Model"),
72
  gr.Textbox(value='{"max_single_segment_time": 15000}', label="VAD Kwargs"),
73
  gr.Slider(1, 10, value=1, step=1, label="Batch Size"),