Update app.py

app.py

@@ -26,6 +26,7 @@ model = AutoModel(
     trust_remote_code=False,
     remote_code="./model.py",
     vad_model="fsmn-vad",
+    spk_model="cam++",
     vad_kwargs={"max_single_segment_time": 15000},
     ncpu=torch.get_num_threads(),
     batch_size=1,
@@ -33,8 +34,8 @@ model = AutoModel(
     device=device,
 )
 
-def transcribe_audio(file_path, vad_model="fsmn-vad", vad_kwargs='{"max_single_segment_time": 15000}', 
-                     batch_size=16, language="auto", use_itn=True, batch_size_s=60, 
+def transcribe_audio(file_path, vad_model="fsmn-vad", spk_model="cam++", vad_kwargs='{"max_single_segment_time": 15000}', 
+                     batch_size=1, language="auto", use_itn=True, batch_size_s=60, 
                      merge_vad=True, merge_length_s=15, batch_size_threshold_s=50, 
                      hotword=" ", ban_emo_unk=True):
     try:
@@ -65,6 +66,7 @@ def transcribe_audio(file_path, vad_model="fsmn-vad", vad_kwargs='{"max_single_s
 inputs = [
     gr.Audio(type="filepath"),
     gr.Textbox(value="fsmn-vad", label="VAD Model"),
+    gr.Textbox(value="cam++", label="SPK Model"),
     gr.Textbox(value='{"max_single_segment_time": 15000}', label="VAD Kwargs"),
     gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
     gr.Textbox(value="auto", label="Language"),