Spaces:
Running
Running
File size: 2,878 Bytes
d317550 e2bed65 d317550 e2bed65 d317550 e2bed65 d317550 e2bed65 d317550 e2bed65 d317550 e2bed65 b7fa700 e2bed65 d317550 b7fa700 e2bed65 b7fa700 d317550 e2bed65 d317550 b7fa700 d317550 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
# Credit to Team UVR : https://github.com/Anjok07/ultimatevocalremovergui
from typing import Optional
import soundfile as sf
import os
import torch
from uvr.models import MDX, Demucs, VrNetwork, MDXC
class MusicSeparator:
def __init__(self,
model_dir: Optional[str] = None,
output_dir: Optional[str] = None):
self.model = None
self.device = self.get_device()
self.available_devices = ["cpu", "cuda"]
self.model_dir = model_dir
self.output_dir = output_dir
self.sample_rate = 16000
self.available_models = ["UVR-MDX-NET-Inst_1", "UVR-MDX-NET-Inst_HQ_1"]
self.default_model = self.available_models[0]
self.current_model_size = self.default_model
self.model_config = {
"segment": 256,
"split": True
}
def update_model(self,
model_name: str = "UVR-MDX-NET-Inst_1",
device: Optional[str] = None,
segment_size: int = 256):
if device is None:
device = self.device
self.device = device
self.model_config = {
"segment": segment_size,
"split": True
}
self.model = MDX(name=model_name,
other_metadata=self.model_config,
device=self.device,
logger=None,
model_dir=self.model_dir)
def separate(self,
audio_file_path: str,
model_name: str,
device: Optional[str] = None,
segment_size: int = 256,
save_file: bool = True):
if device is None:
device = self.device
model_config = {
"segment": segment_size,
"split": True
}
if self.model is None or self.current_model_size != model_name or self.model_config != model_config:
self.update_model(
model_name=model_name,
device=device,
segment_size=segment_size
)
result = self.model(audio_file_path)
instrumental, vocals = result["instrumental"].T, result["vocals"].T
if save_file:
filename, ext = os.path.splitext(audio_file_path)
filename = os.path.basename(filename) + ".wav"
instrumental_output_path = os.path.join(self.output_dir, "instrumental", filename)
vocals_output_path = os.path.join(self.output_dir, "vocals", filename)
sf.write(instrumental_output_path, instrumental, self.sample_rate, format="WAV")
sf.write(vocals_output_path, vocals, self.sample_rate, format="WAV")
return instrumental, vocals
@staticmethod
def get_device():
return "cuda" if torch.cuda.is_available() else "cpu" |