none commited on
Commit
3f053af
·
1 Parent(s): e59985f
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import tempfile
3
 
4
  import gradio as gr
@@ -23,6 +24,9 @@ from src.moviedubber.infer_with_mmlm_result import concat_movie_with_audio, get_
23
  from src.moviedubber.model.utils import convert_char_to_pinyin
24
 
25
 
 
 
 
26
  def load_asr_model(model_id="openai/whisper-large-v3-turbo"):
27
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
28
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
 
1
  import os
2
+ import sys
3
  import tempfile
4
 
5
  import gradio as gr
 
24
  from src.moviedubber.model.utils import convert_char_to_pinyin
25
 
26
 
27
+ sys.path.append("src/third_party/BigVGAN")
28
+
29
+
30
  def load_asr_model(model_id="openai/whisper-large-v3-turbo"):
31
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
32
  model = AutoModelForSpeechSeq2Seq.from_pretrained(