OpenSound commited on
Commit
b43a41d
·
verified ·
1 Parent(s): 5fc4bf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -154,9 +154,10 @@ def tse(gt_file_input, text_input, num_infer_steps, eta, seed, guidance_scale, g
154
  inputs = {key: value.to(device) for key, value in inputs.items()}
155
  timbre = clapmodel.get_text_features(**inputs)
156
 
157
-
158
- pred = sample_diffusion(mixture, timbre, num_infer_steps, eta, seed, guidance_scale, guidance_rescale)
159
- return sample_rate, pred.squeeze().cpu().numpy()
 
160
 
161
 
162
  # CSS styling (optional)
 
154
  inputs = {key: value.to(device) for key, value in inputs.items()}
155
  timbre = clapmodel.get_text_features(**inputs)
156
 
157
+ mixture = autoencoder(embedding=mixture).squeeze(1)
158
+ # pred = sample_diffusion(mixture, timbre, num_infer_steps, eta, seed, guidance_scale, guidance_rescale)
159
+ # return sample_rate, pred.squeeze().cpu().numpy()
160
+ return sample_rate, mixture.squeeze().cpu().numpy()
161
 
162
 
163
  # CSS styling (optional)