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",
+    model="dx2102/llama-midi-2",
     # 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\n')[-1]
+    txt = txt.split('---\n')[-1]
     # track = symusic.core.TrackSecond()
     tracks = {}