jhtonyKoo commited on
Commit
07ebfd5
·
1 Parent(s): 93382c1

modify app

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. modules/loss.py +0 -1
app.py CHANGED
@@ -203,7 +203,9 @@ with gr.Blocks() as demo:
203
  "y_lim": [y_min, y_max],
204
  }
205
 
206
- return final_audio, final_params, final_log, loss_df
 
 
207
 
208
  ito_button.click(
209
  run_ito,
 
203
  "y_lim": [y_min, y_max],
204
  }
205
 
206
+ # Return the plot configuration along with the data
207
+ return final_audio, final_params, final_log, (plot_config, loss_df)
208
+
209
 
210
  ito_button.click(
211
  run_ito,
modules/loss.py CHANGED
@@ -164,7 +164,6 @@ class Loss:
164
  self.multi_scale_spectral_midside = MultiScale_Spectral_Loss_MidSide_DDSP(mode='midside', eps=args.eps, device=device)
165
  self.multi_scale_spectral_ori = MultiScale_Spectral_Loss_MidSide_DDSP(mode='ori', eps=args.eps, device=device)
166
  self.gain = RMSLoss(reduce=reduce)
167
- self.infonce = infoNCE
168
  # perceptual weighting with mel scaled spectrograms
169
  self.mrs_mel_perceptual = auraloss.freq.MultiResolutionSTFTLoss(
170
  fft_sizes=[1024, 2048, 8192],
 
164
  self.multi_scale_spectral_midside = MultiScale_Spectral_Loss_MidSide_DDSP(mode='midside', eps=args.eps, device=device)
165
  self.multi_scale_spectral_ori = MultiScale_Spectral_Loss_MidSide_DDSP(mode='ori', eps=args.eps, device=device)
166
  self.gain = RMSLoss(reduce=reduce)
 
167
  # perceptual weighting with mel scaled spectrograms
168
  self.mrs_mel_perceptual = auraloss.freq.MultiResolutionSTFTLoss(
169
  fft_sizes=[1024, 2048, 8192],