TaiYouWeb commited on
Commit
59fa2b8
·
verified ·
1 Parent(s): 3830518

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,7 @@ model = AutoModel(
38
  device=device,
39
  )
40
 
41
- def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", vad_kwargs='{"max_single_segment_time": 30000}',
42
  batch_size=1, language="auto", use_itn=True, batch_size_s=60,
43
  merge_vad=True, merge_length_s=15, batch_size_threshold_s=50,
44
  hotword=" ", ban_emo_unk=True):
@@ -57,7 +57,6 @@ def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", vad_
57
  merge_length_s=merge_length_s,
58
  batch_size_threshold_s=batch_size_threshold_s,
59
  hotword=hotword,
60
- spk_model=spk_model,
61
  ban_emo_unk=ban_emo_unk
62
  )
63
 
@@ -72,7 +71,8 @@ inputs = [
72
  gr.Audio(type="filepath"),
73
  gr.Textbox(value="fsmn-vad", label="VAD Model"),
74
  gr.Textbox(value="ct-punc", label="PUNC Model"),
75
- gr.Textbox(value='{"max_single_segment_time": 30000}', label="VAD Kwargs"),
 
76
  gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
77
  gr.Textbox(value="auto", label="Language"),
78
  gr.Checkbox(value=True, label="Use ITN"),
 
38
  device=device,
39
  )
40
 
41
+ def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", spk_model="cam++", vad_kwargs='{"max_single_segment_time": 15000}',
42
  batch_size=1, language="auto", use_itn=True, batch_size_s=60,
43
  merge_vad=True, merge_length_s=15, batch_size_threshold_s=50,
44
  hotword=" ", ban_emo_unk=True):
 
57
  merge_length_s=merge_length_s,
58
  batch_size_threshold_s=batch_size_threshold_s,
59
  hotword=hotword,
 
60
  ban_emo_unk=ban_emo_unk
61
  )
62
 
 
71
  gr.Audio(type="filepath"),
72
  gr.Textbox(value="fsmn-vad", label="VAD Model"),
73
  gr.Textbox(value="ct-punc", label="PUNC Model"),
74
+ gr.Textbox(value="cam++", label="SPK Model"),
75
+ gr.Textbox(value='{"max_single_segment_time": 15000}', label="VAD Kwargs"),
76
  gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
77
  gr.Textbox(value="auto", label="Language"),
78
  gr.Checkbox(value=True, label="Use ITN"),