Update app.py

app.py

@@ -17,7 +17,7 @@ print('\n\n\n')
 print('Loading model...')
 pipe = transformers.pipeline(
     "text-generation", 
-    model="dx2102/llama-midi-2",
+    model="dx2102/llama-midi",
     # revision="c303c108399aba837146e893375849b918f413b3",
     torch_dtype="bfloat16",
     device="cuda",
@@ -50,7 +50,7 @@ example_prefix = '''pitch duration wait velocity instrument
 
 def postprocess(txt, path):
     # remove prefix
-    txt = txt.split('---\n')[-1]
+    txt = txt.split('\n\n')[-1]
     # track = symusic.core.TrackSecond()
     tracks = {}
 
@@ -58,7 +58,7 @@ def postprocess(txt, path):
     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, instrument = line.split(' ')
             pitch, duration, wait, velocity = [int(x) for x in [pitch, duration, wait, velocity]]
             if instrument not in tracks:
                 tracks[instrument] = symusic.core.TrackSecond()