modify app

app.py

@@ -163,7 +163,6 @@ with gr.Blocks() as demo:
                 y_title="Loss",
                 height=300,
                 width=600,
-                y_lim=[None, None],
             )
             ito_log = gr.Textbox(label="ITO Log", lines=10)
 
@@ -188,10 +187,21 @@ with gr.Blocks() as demo:
             final_log = log
             loss_df = loss_data
 
-        # # Update y_lim based on the actual min and max values
-        # y_min = loss_df['loss'].min()
-        # y_max = loss_df['loss'].max()
-        # ito_loss_plot.update(y_lim=[y_min, y_max])
+        # Calculate y_min and y_max
+        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 final_audio, final_params, final_log, loss_df