modify app

app.py

@@ -141,8 +141,10 @@ def perform_ito(input_audio, reference_audio, ito_reference_audio, num_steps, op
 
 
 """ APP display """
+gr.Markdown("# ITO-Master: Inference Time Optimization for Mastering Style Transfer")
+
 with gr.Blocks() as demo:
-    gr.Markdown("# Mastering Style Transfer Demo")
+    gr.Markdown("# Step 1: Mastering Style Transfer")
 
     with gr.Tab("Upload Audio"):
         with gr.Row():
@@ -189,7 +191,7 @@ with gr.Blocks() as demo:
             outputs=[output_audio_yt, param_output_yt, error_message_yt]
         )
 
-    gr.Markdown("## Inference Time Optimization (ITO)")
+    gr.Markdown("## Step 2: Inference Time Optimization (ITO)")
     
     with gr.Row():
         ito_reference_audio = gr.Audio(label="ITO Reference Audio (optional)")
@@ -242,20 +244,8 @@ with gr.Blocks() as demo:
         y_min = loss_df['loss'].min()
         y_max = loss_df['loss'].max()
 
-        # Create a new plot configuration
-        plot_config = {
-            "x": "step",
-            "y": "loss",
-            "title": "ITO Loss Curve",
-            "x_title": "Step",
-            "y_title": "Loss",
-            "height": 300,
-            "width": 600,
-            "y_lim": [y_min, y_max],
-        }
-
         # Return the plot configuration along with the data
-        return final_audio, final_params, final_log, (plot_config, loss_df)
+        return final_audio, final_params, final_log, loss_df
 
 
     ito_button.click(