Spaces:
Runtime error
Runtime error
pengdaqian
commited on
Commit
·
c05202f
1
Parent(s):
c04d3a7
fix
Browse files
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
|
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')
|