fix bug creapy config

app.py

@@ -10,6 +10,7 @@ import librosa
 from pvq_manipulation.helper.vad import EnergyVAD
 import gradio as gr
 from pvq_manipulation.helper.creapy_wrapper import process_file
+from creapy.utils import config
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 pvq_labels = ['Weight', 'Resonance', 'Breathiness', 'Roughness', 'Loudness', 'Strain', 'Pitch']
@@ -34,6 +35,11 @@ normalizing_flow = FFJORD.load_model(storage_dir_normalizing_flow, checkpoint="m
 storage_dir_tts = Path("./models/tts_model/")
 tts_model = Vits_NT.load_model(storage_dir_tts, "model.pt")
 
+
+config._CONFIG_DIR = "./pvq_manipulation/helper/creapy_config.yaml"
+config._USER_CONFIG_DIR = "./pvq_manipulation/helper/user_config.yaml"
+config.USER_CONFIG_DIR = "./pvq_manipulation/helper/user_config.yaml"
+
 # load hubert features model
 hubert_model = HubertExtractor(
     layer=SID_LARGE_LAYER,

pvq_manipulation/helper/creapy_wrapper.py

@@ -7,47 +7,11 @@ import pandas as pd
 from scipy.signal.windows import hann
 from pathlib import Path
 from sklearn.impute import SimpleImputer
-import ruamel.yaml
-from os.path import isfile
 import creapy
 from creapy.feature_extraction.feature_extraction import _cpp, _h1_h2, _jitter, _shimmer, _f0mean, _zcr, _ste
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.neural_network import MLPClassifier
 
-_CONFIG_DIR = "./pvq_manipulation/helper/creapy_config.yaml"
-_USER_CONFIG_DIR = "./pvq_manipulation/helper/user_config.yaml"
-USER_CONFIG_DIR = "./pvq_manipulation/helper/user_config.yaml"
-
-
-def get_config() -> dict:
-    """
-    returns the configuration file as a dictionary
-
-    Returns:
-        dict: the configuration
-    """
-
-    with open(_CONFIG_DIR) as config_file:
-        config: dict = ruamel.yaml.safe_load(config_file.read())
-
-    with open(_USER_CONFIG_DIR) as user_config_file:
-        _user_config: dict = ruamel.yaml.safe_load(user_config_file.read())
-
-    config['USER'] = _user_config
-
-    if not isfile(USER_CONFIG_DIR):
-        return config
-
-    with open(USER_CONFIG_DIR) as user_config_file:
-        user_config: dict = ruamel.yaml.safe_load(user_config_file.read())
-
-    for key in user_config.keys():
-        if key not in config['USER']:
-            raise ValueError(f'Invalid key found in user config: {key}')
-
-    config['USER'].update(user_config)
-    return config
-
 
 class Model:
     """The Model for creaky voice classification."""
@@ -164,7 +128,7 @@ def blockwise_feature_calculation(data: np.ndarray, sr, feature):
 
 
 def process_file(data, sample_rate: int = 16_000):
-    _config = get_config()
+    _config = creapy.utils.get_config()
     user_cfg = _config['USER']
     model_cfg = _config['MODEL']