Update app.py

app.py

@@ -160,6 +160,9 @@ def generate_music(
         threads = []
         segment_outputs = [None] * run_n_segments  # Store outputs in correct order
 
+        start_of_segment = mmtokenizer.tokenize('[start_of_segment]')
+        end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
+
         def process_segment(i, p):
             nonlocal raw_output
             section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')