modify app

app.py

@@ -187,12 +187,12 @@ with gr.Blocks() as demo:
             final_log = log
             loss_df = loss_data
 
-        return final_audio, final_params, final_steps, final_log, loss_df
+        return final_audio, final_params, final_log, loss_df
 
     ito_button.click(
         run_ito,
         inputs=[input_audio, reference_audio, ito_reference_audio, num_steps, optimizer, learning_rate, af_weights],
-        outputs=[ito_output_audio, ito_param_output, ito_steps_taken, ito_log, ito_loss_plot]
+        outputs=[ito_output_audio, ito_param_output, ito_log, ito_loss_plot]
     )
 
 demo.launch()