Update app.py
Browse files
app.py
CHANGED
@@ -326,7 +326,7 @@ def generate_music(
|
|
326 |
return None, None, None
|
327 |
|
328 |
# ------------------ Inference function and Gradio UI ------------------ #
|
329 |
-
def infer(genre_txt_content, lyrics_txt_content, num_segments=
|
330 |
try:
|
331 |
mixed_audio_data, vocal_audio_data, instrumental_audio_data = generate_music(
|
332 |
genre_txt=genre_txt_content,
|
|
|
326 |
return None, None, None
|
327 |
|
328 |
# ------------------ Inference function and Gradio UI ------------------ #
|
329 |
+
def infer(genre_txt_content, lyrics_txt_content, num_segments=4, max_new_tokens=25):
|
330 |
try:
|
331 |
mixed_audio_data, vocal_audio_data, instrumental_audio_data = generate_music(
|
332 |
genre_txt=genre_txt_content,
|