Update extract_feature_print.py

extract_feature_print.py

@@ -46,6 +46,9 @@ def printt(strr):
 
 printt(sys.argv)
 model_path = sys.argv[7]
+if os.path.split(model_path)[0] == "/kaggle/working/Mangio-RVC-Fork/Custom":
+    model_path = "/kaggle/working/Mangio-RVC-Fork/hubert_base.pt"
+    Custom_Embed = True
 
 printt(exp_dir)
 wavPath = "%s/1_16k_wavs" % exp_dir
@@ -69,9 +72,7 @@ def readwave(wav_path, normalize=False):
     feats = feats.view(1, -1)
     return feats
 
-if os.path.split(model_path)[0] == "/kaggle/working/Mangio-RVC-Fork/Custom" == True:
-    model_path = "/kaggle/working/Mangio-RVC-Fork/hubert_base.pt"
-    Custom_Embed = True
+
 # HuBERT model
 printt("load model(s) from {}".format(model_path))
 # if hubert model is exist