TaiYouWeb commited on
Commit
05e5cdb
1 Parent(s): 30c9926

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -26,6 +26,8 @@ model = AutoModel(
26
  trust_remote_code=False,
27
  remote_code="./model.py",
28
  vad_model="fsmn-vad",
 
 
29
  vad_kwargs={"max_single_segment_time": 15000},
30
  ncpu=torch.get_num_threads(),
31
  batch_size=1,
@@ -55,9 +57,7 @@ def transcribe_audio(file_path, vad_model="fsmn-vad", vad_kwargs='{"max_single_s
55
  ban_emo_unk=ban_emo_unk
56
  )
57
 
58
- text = res[0]
59
-
60
- return text
61
 
62
  except Exception as e:
63
  return str(e)
 
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(),
33
  batch_size=1,
 
57
  ban_emo_unk=ban_emo_unk
58
  )
59
 
60
+ return res[0]
 
 
61
 
62
  except Exception as e:
63
  return str(e)