Update app.py

app.py

@@ -48,8 +48,6 @@ example_prefix = '''pitch duration wait velocity instrument
 
 
 def postprocess(txt, path):
-    # assert txt.startswith(prompt)
-    txt = txt.split('\n\n')[-1]
 
     # track = symusic.core.TrackSecond()
     tracks = {}
@@ -57,7 +55,8 @@ def postprocess(txt, path):
     now = 0
     # we need to ignore the invalid output by the model
     try:
-        for line in txt.split('\n'):
+        # Remove the first three lines. 1: Score title, 2: The "pitch duration..." hint, 3: Empty line
+        for line in txt.split('\n')[3:]:
             pitch, duration, wait, velocity, instrument = line.split()
             pitch, duration, wait, velocity = [int(x) for x in [pitch, duration, wait, velocity]]
             if instrument not in tracks: