Update app.py

app.py

@@ -48,15 +48,15 @@ example_prefix = '''pitch duration wait velocity instrument
 
 
 def postprocess(txt, path):
-
+    # remove prefix
+    txt = txt.split('\n\n')[-1]
     # track = symusic.core.TrackSecond()
     tracks = {}
 
     now = 0
-    # we need to ignore the invalid output by the model
-    try:
-        # Remove the first three lines. 1: Score title, 2: The "pitch duration..." hint, 3: Empty line
-        for line in txt.split('\n')[3:]:
+    for line in txt.split('\n'):
+        # we need to ignore the invalid output by the model
+        try:
             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:
@@ -73,8 +73,8 @@ def postprocess(txt, path):
                 velocity=int(velocity * 4),
             ))
             now += wait
-    except Exception as e:
-        print('Postprocess: Ignored error:', e)
+        except Exception as e:
+            print(f'Postprocess: Ignored line: "{line}" because of error:', e)
     
     print(f'Postprocess: Got {sum(len(track.notes) for track in tracks.values())} notes')