Update app.py

app.py

@@ -156,7 +156,8 @@ def tse(gt_file_input, text_input, num_infer_steps, eta, seed, guidance_scale, g
 
     
     pred = sample_diffusion(mixture, timbre, num_infer_steps, eta, seed, guidance_scale, guidance_rescale)
-    return sample_rate, pred.squeeze().cpu().numpy()
+    # return sample_rate, pred.squeeze().cpu().numpy()
+    return sample_rate, mixture
 
 
 # CSS styling (optional)