From 2419d5e1a9f78cf54641e514015a80fae1b6e7f0 Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Wed, 30 Aug 2023 19:02:41 +0800
Subject: [PATCH] Add files via upload

---
 infer/lib/rmvpe.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/infer/lib/rmvpe.py b/infer/lib/rmvpe.py
index 0c288b2..b18c896 100644
--- a/infer/lib/rmvpe.py
+++ b/infer/lib/rmvpe.py
@@ -1,4 +1,4 @@
-import pdb
+import pdb,os
 
 import numpy as np
 import torch
@@ -577,7 +577,7 @@ class RMVPE:
             import onnxruntime as ort
 
             ort_session = ort.InferenceSession(
-                "rmvpe.onnx", providers=["DmlExecutionProvider"]
+                "%s/rmvpe.onnx"%os.environ["rmvpe_root"], providers=["DmlExecutionProvider"]
             )
             self.model = ort_session
         else: