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