Update app.py

app.py

@@ -113,7 +113,7 @@ def load_audio_mono(filepath, sampling_rate=16000):
     return audio
 
 def split_lyrics(lyrics: str):
-    pattern = r"\[(\w+)\](.*?)\n(?=\[|\Z)"
+    pattern = r"\[(\w+)\]\s*(.*?)(?=\s*\n\[|\Z)"
     segments = re.findall(pattern, lyrics, re.DOTALL)
     structured_lyrics = [f"[{seg[0]}]\n{seg[1].strip()}\n\n" for seg in segments]
     return structured_lyrics
@@ -122,7 +122,7 @@ def split_lyrics(lyrics: str):
 def generate_music(
         genre_txt=None,
         lyrics_txt=None,
-        run_n_segments=3,
+        run_n_segments=2,
         max_new_tokens=35,
         use_audio_prompt=False,
         audio_prompt_path="",