Update app.py
Browse files
app.py
CHANGED
@@ -105,7 +105,7 @@ def get_last_mp3_file(output_dir):
|
|
105 |
return mp3_files_with_path[0]
|
106 |
|
107 |
@spaces.GPU
|
108 |
-
def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
|
109 |
# Create temporary files
|
110 |
genre_txt_path = create_temp_file(genre_txt_content, prefix="genre_")
|
111 |
lyrics_txt_path = create_temp_file(lyrics_txt_content, prefix="lyrics_")
|
|
|
105 |
return mp3_files_with_path[0]
|
106 |
|
107 |
@spaces.GPU
|
108 |
+
def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=200):
|
109 |
# Create temporary files
|
110 |
genre_txt_path = create_temp_file(genre_txt_content, prefix="genre_")
|
111 |
lyrics_txt_path = create_temp_file(lyrics_txt_content, prefix="lyrics_")
|