add config

app.py

@@ -2,22 +2,20 @@ import gradio as gr
 import torch
 import soundfile as sf
 import numpy as np
+import yaml
 from inference import MasteringStyleTransfer
 from utils import download_youtube_audio
 from config import config
 
 # Initialize MasteringStyleTransfer
 args = type('Args', (), vars(config))()
-mastering_transfer = MasteringStyleTransfer(args)
+# load network configurations
+with open(args.path_to_config, 'r') as f:
+    configs = yaml.full_load(f)
+args.cfg_converter = configs['TCN']['param_mapping']
+args.cfg_enc = configs['Effects_Encoder']['default']
 
-# # Initialize MasteringStyleTransfer
-# args = type('Args', (), {
-#     "model_path": "models/mastering_converter.pt",
-#     "encoder_path": "models/effects_encoder.pt",
-#     "sample_rate": 44100,
-#     "path_to_config": "networks/configs.yaml"
-# })()
-# mastering_transfer = MasteringStyleTransfer(args)
+mastering_transfer = MasteringStyleTransfer(args)
 
 def process_audio(input_audio, reference_audio, perform_ito):
     # Process the audio files

config.py

@@ -24,7 +24,5 @@ class Config:
     train_ito = True
     max_iter_ito = 101
     ito_type = "blackbox"
-    
-    # You can add more configurations as needed
 
-config = Config()
+config = Config()