modify app

app.py

@@ -10,9 +10,11 @@ from config import args
 mastering_transfer = MasteringStyleTransfer(args)
 
 def process_audio(input_audio, reference_audio):
+    print("before style transfer")
     output_audio, predicted_params, _, _, _, sr = mastering_transfer.process_audio(
         input_audio, reference_audio, reference_audio, {}, False
     )
+    print("style transfer completed")
     
     param_output = mastering_transfer.get_param_output_string(predicted_params)
     
@@ -26,6 +28,7 @@ def process_audio(input_audio, reference_audio):
     elif output_audio.ndim > 2:
         output_audio = output_audio.squeeze()
     
+    print("returning result")
     return (sr, output_audio), param_output
 
 def perform_ito(input_audio, reference_audio, ito_reference_audio, num_steps, optimizer, learning_rate, af_weights):
@@ -73,7 +76,7 @@ with gr.Blocks() as demo:
             input_audio = gr.Audio(label="Input Audio")
             reference_audio = gr.Audio(label="Reference Audio")
         
-        process_button = gr.Button("Process")
+        process_button = gr.Button("Process Mastering Style Transfer")
         
         with gr.Row():
             output_audio = gr.Audio(label="Output Audio")