TaiYouWeb commited on
Commit
323958e
1 Parent(s): 3672564

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -10,9 +10,6 @@ import tempfile
10
  import json
11
  from typing import Optional
12
 
13
- from pyannote.audio import Audio, Pipeline
14
- from pyannote.core import Segment
15
-
16
  import torch
17
  import gradio as gr
18
 
@@ -28,8 +25,8 @@ model = AutoModel(
28
  model=model_dir,
29
  trust_remote_code=False,
30
  remote_code="./model.py",
31
- vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
32
- punc_model="ct-punc",
33
  spk_model="cam++",
34
  vad_kwargs={"max_single_segment_time": 15000},
35
  ncpu=torch.get_num_threads(),
 
10
  import json
11
  from typing import Optional
12
 
 
 
 
13
  import torch
14
  import gradio as gr
15
 
 
25
  model=model_dir,
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(),