modify app

app.py

@@ -11,7 +11,7 @@ mastering_transfer = MasteringStyleTransfer(args)
 
 def process_audio(input_audio, reference_audio, perform_ito, ito_reference_audio=None):
     # Process the audio files
-    output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, sr, _ = mastering_transfer.process_audio(
+    output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, sr = mastering_transfer.process_audio(
         input_audio, reference_audio, ito_reference_audio if ito_reference_audio else reference_audio, {}, perform_ito
     )
     
@@ -33,7 +33,7 @@ def process_youtube_with_ito(input_url, reference_url, perform_ito, use_same_ref
     reference_audio = download_youtube_audio(reference_url)
     ito_ref = reference_audio if use_same_reference else download_youtube_audio(ito_reference_url)
     
-    output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, sr, _ = mastering_transfer.process_audio(
+    output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, sr = mastering_transfer.process_audio(
         input_audio, reference_audio, ito_ref, {}, perform_ito, log_ito=True
     )
     

inference.py

@@ -172,7 +172,7 @@ class MasteringStyleTransfer:
             ito_predicted_params = None
             ito_log = None
 
-        return output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, self.args.sample_rate, duration
+        return output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, self.args.sample_rate
 
     def print_param_difference(self, initial_params, ito_params):
         all_diffs = []