Update app.py

app.py

@@ -104,7 +104,7 @@ except Exception as e:
 LYRICS_PATTERN = re.compile(r"\[(\w+)\](.*?)\n(?=\[|\Z)", re.DOTALL)
 
 # ------------------ GPU decorated generation function ------------------ #
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=175)
 def generate_music(
     max_new_tokens=5,
     run_n_segments=2,
@@ -326,7 +326,7 @@ def generate_music(
                 return None, None, None
 
 # ------------------ Inference function and Gradio UI ------------------ #
-def infer(genre_txt_content, lyrics_txt_content, num_segments=1, max_new_tokens=25):
+def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=25):
     try:
         mixed_audio_data, vocal_audio_data, instrumental_audio_data = generate_music(
             genre_txt=genre_txt_content,