modify app

app.py

@@ -203,7 +203,9 @@ with gr.Blocks() as demo:
             "y_lim": [y_min, y_max],
         }
 
-        return final_audio, final_params, final_log, loss_df
+        # Return the plot configuration along with the data
+        return final_audio, final_params, final_log, (plot_config, loss_df)
+
 
     ito_button.click(
         run_ito,

modules/loss.py

@@ -164,7 +164,6 @@ class Loss:
         self.multi_scale_spectral_midside = MultiScale_Spectral_Loss_MidSide_DDSP(mode='midside', eps=args.eps, device=device)
         self.multi_scale_spectral_ori = MultiScale_Spectral_Loss_MidSide_DDSP(mode='ori', eps=args.eps, device=device)
         self.gain = RMSLoss(reduce=reduce)
-        self.infonce = infoNCE
         # perceptual weighting with mel scaled spectrograms
         self.mrs_mel_perceptual = auraloss.freq.MultiResolutionSTFTLoss(
             fft_sizes=[1024, 2048, 8192],