modify app

app.py

@@ -145,9 +145,10 @@ def perform_ito(input_audio, reference_audio, ito_reference_audio, num_steps, op
         yield (args.sample_rate, current_output), ito_param_output, result['step'], ito_log, pd.DataFrame(loss_values), all_results
 
 def update_ito_output(all_results, selected_step):
-    print(all_results)
+    print(all_results[selected_step - 1])
     print(selected_step)
     selected_result = all_results[selected_step - 1]
+    print(selected_result['audio'].shape)
     return (args.sample_rate, selected_result['audio']), selected_result['params'], selected_result['log']
 
 

inference.py

@@ -159,7 +159,7 @@ class MasteringStyleTransfer:
             all_results.append({
                 'step': step + 1,
                 'loss': total_loss.item(),
-                'audio': output_audio.detach(),
+                'audio': output_audio.detach().cpu().numpy(),
                 'params': current_params,
                 'log': log_entry
             })