modify app

app.py

@@ -156,6 +156,8 @@ def run_ito(input_audio, reference_audio, ito_reference_audio, num_steps, optimi
     return all_results, min_loss_step, loss_df
 
 def update_ito_output(all_results, selected_step):
+    print(all_results)
+    print(selected_step)
     selected_result = all_results[selected_step]
     return (args.sample_rate, selected_result['audio']), selected_result['params'], selected_result['log']
 

inference.py

@@ -167,7 +167,7 @@ class MasteringStyleTransfer:
             total_loss.backward()
             optimizer.step()
 
-            yield all_results[-1]
+            # yield all_results[-1]
 
         return all_results, min_loss_step