pengdaqian commited on
Commit
c05202f
·
1 Parent(s): c04d3a7
Files changed (2) hide show
  1. app.py +1 -4
  2. warm.py +5 -0
app.py CHANGED
@@ -3,19 +3,17 @@ import threading
3
 
4
  os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
5
 
6
- from music.search import get_random_spit, get_albums, search_youtube, download_youtube, get_youtube, download_random
7
  from vits.models import SynthesizerInfer
8
  import whisper.inference
9
  from omegaconf import OmegaConf
10
  import torchcrepe
11
  import torch
12
- import io
13
  import gradio as gr
14
  import librosa
15
  import numpy as np
16
  import soundfile
17
  from spleeter.separator import Separator
18
- from spleeter.audio.adapter import AudioAdapter
19
  from pydub import AudioSegment
20
  import uuid
21
 
@@ -86,7 +84,6 @@ load_svc_model("vits_pretrain/sovits5.0-48k-debug.pth", model)
86
  model.eval()
87
  model.to(device)
88
  separator = Separator('spleeter:2stems')
89
- audio_loader = AudioAdapter.default()
90
  whisper_model = whisper.inference.load_model(os.path.join("whisper_pretrain", "medium.pt"))
91
 
92
  # warm up
 
3
 
4
  os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
5
 
6
+ from music.search import get_youtube, download_random
7
  from vits.models import SynthesizerInfer
8
  import whisper.inference
9
  from omegaconf import OmegaConf
10
  import torchcrepe
11
  import torch
 
12
  import gradio as gr
13
  import librosa
14
  import numpy as np
15
  import soundfile
16
  from spleeter.separator import Separator
 
17
  from pydub import AudioSegment
18
  import uuid
19
 
 
84
  model.eval()
85
  model.to(device)
86
  separator = Separator('spleeter:2stems')
 
87
  whisper_model = whisper.inference.load_model(os.path.join("whisper_pretrain", "medium.pt"))
88
 
89
  # warm up
warm.py ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ from spleeter.separator import Separator
2
+
3
+ separator = Separator('spleeter:2stems')
4
+ # warm up
5
+ separator.separate_to_file('warm.wav', '/tmp/warm')