src/constants.py

# Third-party
import torch

# Global Variables
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
ONNX_MODEL_PATH = "./pretrained_models/MDX_net/Kim_Vocal.onnx"
INPUT_FOLDER = "./datasets/input"
OUTPUT_FOLDER = "./datasets/output"