KingNish commited on
Commit
848a314
·
verified ·
1 Parent(s): d610f33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -510,7 +510,7 @@ def generate_music(
510
  # return final mix, upsampled vocal stem, upsampled instrumental stem
511
  return (44100, (mix_output.cpu().numpy() * 32767).astype(np.int16)), (44100, (vocal_output.cpu().numpy() * 32767).astype(np.int16)), (44100, (instrumental_output.cpu().numpy() * 32767).astype(np.int16))
512
 
513
- def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=10):
514
  # Execute the command
515
  try:
516
  mixed_audio_data, vocal_audio_data, instrumental_audio_data = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments,
 
510
  # return final mix, upsampled vocal stem, upsampled instrumental stem
511
  return (44100, (mix_output.cpu().numpy() * 32767).astype(np.int16)), (44100, (vocal_output.cpu().numpy() * 32767).astype(np.int16)), (44100, (instrumental_output.cpu().numpy() * 32767).astype(np.int16))
512
 
513
+ def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=5):
514
  # Execute the command
515
  try:
516
  mixed_audio_data, vocal_audio_data, instrumental_audio_data = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments,