Choultion-Rudas commited on
Commit
fe02ba0
·
verified ·
1 Parent(s): 3e80b26

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,13 +6,13 @@ import re
6
  from funasr import AutoModel
7
  from funasr.utils.postprocess_utils import rich_transcription_postprocess
8
 
 
9
  model = AutoModel(
10
- model="FunAudioLLM/SenseVoiceSmall",
11
  vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
12
  vad_kwargs={"max_single_segment_time": 30000},
13
- hub="hf",
14
  trust_remote_code=True,
15
- device="cuda:0" if torch.cuda.is_available() else "cpu",
16
  )
17
 
18
 
 
6
  from funasr import AutoModel
7
  from funasr.utils.postprocess_utils import rich_transcription_postprocess
8
 
9
+ device = "cuda:0" if torch.cuda.is_available() else "cpu"
10
  model = AutoModel(
11
+ model="iic/SenseVoiceSmall",
12
  vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
13
  vad_kwargs={"max_single_segment_time": 30000},
 
14
  trust_remote_code=True,
15
+ device=device,
16
  )
17
 
18