Update app.py
Browse files
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=
|
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,
|