Update app.py
Browse files
app.py
CHANGED
@@ -303,7 +303,7 @@ def stage2_inference(model_stage2, stage1_output_set, stage2_output_dir, batch_s
|
|
303 |
return stage2_result
|
304 |
|
305 |
# ----------------------- Main Generation Function (Stage1 + Stage2) -----------------------
|
306 |
-
@spaces.GPU(duration=
|
307 |
def generate_music(
|
308 |
genre_txt="",
|
309 |
lyrics_txt="",
|
|
|
303 |
return stage2_result
|
304 |
|
305 |
# ----------------------- Main Generation Function (Stage1 + Stage2) -----------------------
|
306 |
+
@spaces.GPU(duration=150)
|
307 |
def generate_music(
|
308 |
genre_txt="",
|
309 |
lyrics_txt="",
|