Update extract_feature_print.py

extract_feature_print.py

@@ -74,8 +74,10 @@ models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task(
     [model_path],
     suffix="",
 )
-print(models," | ", saved_cfg," | ", task)
-model = models[0]
+if os.path.split(model_path)[0] == "/kaggle/working/Mangio-RVC-Fork/Custom" == False:
+    model = models[0]
+else:
+    model = HubertModelWithFinalProj.from_pretrained(model_path)
 model = model.to(device)
 printt("move model to %s" % device)
 if device not in ["mps", "cpu"]: