Update app.py

app.py

@@ -231,6 +231,8 @@ def generate_music(
                 else:
                     raw_output = output_seq
 
+        generator()
+        
         # save raw output and check sanity
         ids = raw_output[0].cpu().numpy()
         soa_idx = np.where(ids == mmtokenizer.soa)[0].tolist()