modify app

app.py

@@ -194,9 +194,10 @@ with gr.Blocks() as demo:
             final_log = log
             loss_values = losses
 
-        loss_plot = plot_loss_curve(loss_values)
+        # loss_plot = plot_loss_curve(loss_values)
+        print(loss_values)
         
-        return final_audio, final_params, final_steps, final_log, loss_plot
+        return final_audio, final_params, final_steps, final_log, loss_values
 
     ito_button.click(
         run_ito,