Update app.py

app.py

@@ -149,8 +149,6 @@ def generate_music(
 
         genres = genre_txt.strip()
         lyrics = split_lyrics(lyrics_txt + "\n")
-        print(lyrics)
-        print(len(lyrics))
         # instruction
         full_lyrics = "\n".join(lyrics)
         prompt_texts = [f"Generate music from the given lyrics segment by segment.\n[Genre] {genres}\n{full_lyrics}"]
@@ -167,7 +165,7 @@ def generate_music(
         end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
 
         # Format text prompt
-        run_n_segments = min(run_n_segments + 1, len(lyrics))
+        run_n_segments = min(run_n_segments, len(lyrics)) + 1
 
         print(list(enumerate(tqdm(prompt_texts[:run_n_segments]))))