Update app.py

app.py

@@ -23,7 +23,7 @@ in_space = os.getenv("SYSTEM") == "spaces"
 # =================================================================================================
                        
 @spaces.GPU
-def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
+def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type):
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     start_time = reqtime.time()
@@ -31,7 +31,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
     print('Loading model...')
 
     SEQ_LEN = 8192 # Models seq len
-    PAD_IDX = 767 # Models pad index
+    PAD_IDX = 707 # Models pad index
     DEVICE = 'cuda' # 'cuda'
 
     # instantiate the model
@@ -50,7 +50,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
     print('Loading model checkpoint...')
 
     model.load_state_dict(
-        torch.load('Ultimate_Accompaniment_Transformer_Small_Improved_Trained_Model_13649_steps_0.3229_loss_0.898_acc.pth',
+        torch.load('Chords_Progressions_Transformer_Small_2048_Trained_Model_12947_steps_0.9316_loss_0.7386_acc.pth',
                    map_location=DEVICE))
     print('=' * 70)
 
@@ -369,21 +369,21 @@ if __name__ == "__main__":
    
     app = gr.Blocks()
     with app:
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Ultimate Accompaniment Transformer</h1>")
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Generate unique accompaniment for any melody</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Chords Progressions Transformer</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Chords-conditioned music transformer</h1>")
         gr.Markdown(
-            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Ultimate-Accompaniment-Transformer&style=flat)\n\n"
-            "Accompaniment generation for any monophonic melody\n\n"
-            "Check out [Ultimate Drums Transformer](https://github.com/asigalov61/Ultimate-Accompaniment-Transformer) on GitHub!\n\n"
+            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Chords-Progressions-Transformer&style=flat)\n\n"
+            "Generate music based on chords progressions\n\n"
+            "Check out [Chords Progressions Transformer](https://github.com/asigalov61/Chords-Progressions-Transformer) on GitHub!\n\n"
             "[Open In Colab]"
-            "(https://colab.research.google.com/github/asigalov61/Ultimate-Accompaniment-Transformer/blob/main/Ultimate_Accompaniment_Transformer.ipynb)"
+            "(https://colab.research.google.com/github/asigalov61/Chords-Progressions-Transformer/blob/main/Chords_Progressions_Transformer.ipynb)"
             " for faster execution and endless generation"
         )
         gr.Markdown("## Upload your MIDI or select a sample example MIDI")
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
-        input_num_tokens = gr.Slider(4, 128, value=32, step=1, label="Number of composition chords to generate accompaniment for")
-        input_acc_type = gr.Checkbox(label='Force accompaniment generation for each melody note')
+        input_num_tokens = gr.Slider(4, 128, value=32, step=1, label="Number of composition chords to generate progression for")
+        input_conditioning_type = gr.Radio(["Chords", "Chords-Times", "Chords-Times-Durations"], label="Conditioning type")
         
         run_btn = gr.Button("generate", variant="primary")
 
@@ -396,18 +396,24 @@ if __name__ == "__main__":
         output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
 
 
-        run_event = run_btn.click(GenerateAccompaniment, [input_midi, input_num_tokens, input_acc_type],
+        run_event = run_btn.click(GenerateAccompaniment, [input_midi, input_num_tokens, input_conditioning_type],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Ultimate-Accompaniment-Transformer-Melody-Seed-1.mid", 128, True], 
-             ["Ultimate-Accompaniment-Transformer-Melody-Seed-2.mid", 128, False], 
-             ["Ultimate-Accompaniment-Transformer-Melody-Seed-3.mid", 128, True],
-             ["Ultimate-Accompaniment-Transformer-Melody-Seed-4.mid", 128, False],
-             ["Ultimate-Accompaniment-Transformer-Melody-Seed-5.mid", 128, True],
-             ["Ultimate-Accompaniment-Transformer-Melody-Seed-6.mid", 128, False],
-             ["Ultimate-Accompaniment-Transformer-Melody-Seed-7.mid", 128, True]],
-            [input_midi, input_num_tokens, input_acc_type],
+            [["Chords-Progressions-Transformer-Piano-Seed-1.mid", 128, "Chords"], 
+             ["Chords-Progressions-Transformer-Piano-Seed-2.mid", 128, "Chords"], 
+             ["Chords-Progressions-Transformer-Piano-Seed-3.mid", 128, "Chords"],
+             ["Chords-Progressions-Transformer-Piano-Seed-4.mid", 128, "Chords"],
+             ["Chords-Progressions-Transformer-Piano-Seed-5.mid", 128, "Chords"],
+             ["Chords-Progressions-Transformer-Piano-Seed-6.mid", 128, "Chords"],
+             ["Chords-Progressions-Transformer-MI-Seed-1.mid", 128, "Chords"]
+             ["Chords-Progressions-Transformer-MI-Seed-2.mid", 128, "Chords"]
+             ["Chords-Progressions-Transformer-MI-Seed-3.mid", 128, "Chords"]
+             ["Chords-Progressions-Transformer-MI-Seed-4.mid", 128, "Chords"]
+             ["Chords-Progressions-Transformer-MI-Seed-5.mid", 128, "Chords"]
+             ["Chords-Progressions-Transformer-MI-Seed-6.mid", 128, "Chords"]
+            ],
+            [input_midi, input_num_tokens, input_conditioning_type],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
             GenerateAccompaniment,
             cache_examples=True,