Spaces:
Runtime error
Runtime error
Jarod Castillo
commited on
Commit
·
d24a3d9
1
Parent(s):
4d09908
removed pip from requirements, fixed pretrained model pathing
Browse files- requirements.txt +0 -1
- src/constants.py +1 -1
- src/infer.py +1 -1
requirements.txt
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
pip~=23.2.1
|
| 2 |
torch~=2.0.1
|
| 3 |
onnxruntime~=1.15.1
|
| 4 |
librosa~=0.10.0.post2
|
|
|
|
|
|
|
| 1 |
torch~=2.0.1
|
| 2 |
onnxruntime~=1.15.1
|
| 3 |
librosa~=0.10.0.post2
|
src/constants.py
CHANGED
|
@@ -4,6 +4,6 @@ import torch
|
|
| 4 |
# Global Variables
|
| 5 |
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 6 |
EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
|
| 7 |
-
ONNX_MODEL_PATH = "./
|
| 8 |
INPUT_FOLDER = "./datasets/input"
|
| 9 |
OUTPUT_FOLDER = "./datasets/output"
|
|
|
|
| 4 |
# Global Variables
|
| 5 |
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 6 |
EXECUTION_PROVIDER_LIST = ["CUDAExecutionProvider", "CPUExecutionProvider"]
|
| 7 |
+
ONNX_MODEL_PATH = "./pretrained_models/MDX_Net_Models/Kim_Vocal.onnx"
|
| 8 |
INPUT_FOLDER = "./datasets/input"
|
| 9 |
OUTPUT_FOLDER = "./datasets/output"
|
src/infer.py
CHANGED
|
@@ -10,7 +10,7 @@ import onnxruntime as ort
|
|
| 10 |
from models.MDX_net.mdx_net import Conv_TDF_net_trimm
|
| 11 |
from loader import Loader
|
| 12 |
|
| 13 |
-
vocal_path =
|
| 14 |
|
| 15 |
# Global Variables
|
| 16 |
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 10 |
from models.MDX_net.mdx_net import Conv_TDF_net_trimm
|
| 11 |
from loader import Loader
|
| 12 |
|
| 13 |
+
vocal_path = "./datasets/output/vocals.wav"
|
| 14 |
|
| 15 |
# Global Variables
|
| 16 |
COMPUTATION_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|