KingNish commited on
Commit
f24e46c
·
verified ·
1 Parent(s): 3037e6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -211,7 +211,7 @@ def generate_music(genre_txt, lyrics_txt, max_new_tokens=5, run_n_segments=2, us
211
  decoded_instrumentals = codec_model.decode(torch.as_tensor(instrumentals.astype(np.int16), dtype=torch.long).unsqueeze(0).permute(1, 0, 2).to(device)).cpu().squeeze(0)
212
 
213
  mixed_audio = (decoded_vocals + decoded_instrumentals) / 2
214
- return (16000, mixed_audio.numpy())
215
 
216
  @spaces.GPU(duration=120)
217
  def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=10):
 
211
  decoded_instrumentals = codec_model.decode(torch.as_tensor(instrumentals.astype(np.int16), dtype=torch.long).unsqueeze(0).permute(1, 0, 2).to(device)).cpu().squeeze(0)
212
 
213
  mixed_audio = (decoded_vocals + decoded_instrumentals) / 2
214
+ return (16000, mixed_audio.tensor.detach().numpy())
215
 
216
  @spaces.GPU(duration=120)
217
  def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=10):