游雁 commited on
Commit
b03989f
1 Parent(s): d737fb7
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -19,6 +19,7 @@ model = AutoModel(model=model,
19
  vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
20
  vad_kwargs={"max_single_segment_time": 30000},
21
  hub="hf",
 
22
  )
23
 
24
  import re
 
19
  vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
20
  vad_kwargs={"max_single_segment_time": 30000},
21
  hub="hf",
22
+ device="cuda"
23
  )
24
 
25
  import re