TaiYouWeb commited on
Commit
f9b643b
·
verified ·
1 Parent(s): ce48894

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -15,14 +15,11 @@ import gradio as gr
15
 
16
  from config import model_config
17
 
18
-
19
-
20
-
21
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
22
  model_dir = snapshot_download(model_config['model_dir'])
23
 
24
  model = AutoModel(
25
- model="iic/SenseVoiceSmall",
26
  trust_remote_code=False,
27
  remote_code="./model.py",
28
  vad_model="fsmn-vad",
 
15
 
16
  from config import model_config
17
 
 
 
 
18
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
19
  model_dir = snapshot_download(model_config['model_dir'])
20
 
21
  model = AutoModel(
22
+ model=model_dir,
23
  trust_remote_code=False,
24
  remote_code="./model.py",
25
  vad_model="fsmn-vad",