Upload main.py

main.py

@@ -0,0 +1,31 @@
+from model import PopMusicTransformer
+import os
+os.environ['CUDA_VISIBLE_DEVICES'] = '0'
+
+def main():
+    # declare model
+    model = PopMusicTransformer(
+        checkpoint='REMI-tempo-checkpoint',
+        is_training=False)
+    
+    # generate from scratch
+    model.generate(
+        n_target_bar=16,
+        temperature=1.2,
+        topk=5,
+        output_path='./result/from_scratch.midi',
+        prompt=None)
+    
+    # generate continuation
+    model.generate(
+        n_target_bar=16,
+        temperature=1.2,
+        topk=5
+        output_path='./result/continuation.midi',
+        prompt='./data/evaluation/000.midi')
+    
+    # close model
+    model.close()
+
+if __name__ == '__main__':
+    main()