Update app.py

app.py

@@ -21,15 +21,16 @@ model = AutoModel(
     trust_remote_code=False,
     remote_code="./model.py",
     vad_model="fsmn-vad",
+    punc_model="ct-punc", 
     vad_kwargs={"max_single_segment_time": 30000},
-    ncpu=4,
+    ncpu=torch.get_num_threads(),
     batch_size=1,
-    hub="ms",
+    hub="hf",
     device=device,
 )
 
-def transcribe_audio(file_path, vad_model="fsmn-vad", vad_kwargs='{"max_single_segment_time": 30000}', 
-                     ncpu=4, batch_size=1, language="auto", use_itn=True, batch_size_s=60, 
+def transcribe_audio(file_path, vad_model="fsmn-vad", punc_model="ct-punc", vad_kwargs='{"max_single_segment_time": 30000}', 
+                     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=" ", spk_model="cam++", ban_emo_unk=False):
     try:
@@ -61,8 +62,8 @@ 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="ct-punc", label="PUNC Model"),
     gr.Textbox(value='{"max_single_segment_time": 30000}', label="VAD Kwargs"),
-    gr.Slider(1, 8, value=4, step=1, label="NCPU"),
     gr.Slider(1, 10, value=1, step=1, label="Batch Size"),
     gr.Textbox(value="auto", label="Language"),
     gr.Checkbox(value=True, label="Use ITN"),