vocal_isolation/constants.py

# Standard
import os

# Third-party
import torch

# Global Variables
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
ONNX_MODEL_PATH = "./vocal_isolation/pretrained_models/Kim_Vocal.onnx"
PRETRAINED_MODELS_DIRECTORY = "./vocal_isolation/pretrained_models"
INPUT_FOLDER = "./dl_audio"
OUTPUT_FOLDER = "./output/mdx_net_kim_vocal/audio"