src/infer.py

# Standard Library Imports
import os
import subprocess

# Third Party Imports
import torch
import onnxruntime as ort

# Local Imports
from models.MDX_net.mdx_net import Conv_TDF_net_trimm
from loader import Loader

vocal_path = r"./datasets/output/vocals.wav"

# Global Variables
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"


def main():
    print(COMPUTATION_DEVICE)