add config

app.py

@@ -4,16 +4,21 @@ import soundfile as sf
 import numpy as np
 from inference import MasteringStyleTransfer
 from utils import download_youtube_audio
+from config import config
 
 # 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"
-})()
+args = type('Args', (), vars(config))()
 mastering_transfer = MasteringStyleTransfer(args)
 
+# # 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)
+
 def process_audio(input_audio, reference_audio, perform_ito):
     # Process the audio files
     output_audio, predicted_params, ito_output_audio, ito_predicted_params, _, sr, _ = mastering_transfer.process_audio(

config.py

@@ -0,0 +1,30 @@
+import os
+
+class Config:
+    model_path = "models/mastering_converter.pt"
+    encoder_path = "models/effects_encoder.pt"
+    sample_rate = 44100
+    path_to_config = "networks/configs.yaml"
+    
+    # Add more configurations as needed
+    inference_only = True
+    evaluate_only = False
+    reload_enc = True
+    reload_converter = True
+    manual_reload_enc = False
+    manual_reload_converter = False
+    manual_reload_name_converter = "dasp_tcn_tuneenc_daspman_loudnessnorm"
+    manual_reload_name_encoder = "dasp_tcn_tuneenc_daspman_loudnessnorm"
+    manual_reload_enc_path = "models/effects_encoder.pt"
+    reload_epoch_converter = 1000
+    reload_epoch_enc = 1000
+    
+    # Add other configurations from the main.py file
+    random_seed = 111
+    train_ito = True
+    max_iter_ito = 101
+    ito_type = "blackbox"
+    
+    # You can add more configurations as needed
+
+config = Config()