Spaces:
Running
Running
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"
|