Spaces:
Running
Running
modify app
Browse files- app.py +3 -1
- 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 |
-
|
|
|
|
|
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],
|