Update app.py

app.py

@@ -118,6 +118,8 @@ def split_lyrics(lyrics: str):
     structured_lyrics = [f"[{seg[0]}]\n{seg[1].strip()}\n\n" for seg in segments]
     return structured_lyrics
 
+
+@spaces.GPU(duration=178)
 def generate_music(
         genre_txt=None,
         lyrics_txt=None,
@@ -167,7 +169,6 @@ def generate_music(
         # Format text prompt
         run_n_segments = min(run_n_segments, len(lyrics)) + 1
 
-        @spaces.GPU(duration=178)
         def generator():
             for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
                 section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')