Update app.py

app.py

@@ -115,7 +115,7 @@ def Harmonize_Melody(input_src_midi,
     
     model = AutoregressiveWrapper(model, ignore_index = PAD_IDX, pad_value=PAD_IDX)
     
-    model_path = 'Melody_Harmonizer_Transformer_Trained_Model_7522_steps_0.6545_loss_0.7906_acc.pth'
+    model_path = 'Melody_Harmonizer_Transformer_Trained_Model_14961_steps_0.4155_loss_0.8664_acc.pth'
     
     model.load_state_dict(torch.load(model_path))
     
@@ -158,7 +158,7 @@ def Harmonize_Melody(input_src_midi,
                               1,
                               filter_logits_fn=top_k,
                               filter_kwargs={'k': model_top_k_sampling_value},
-                              temperature=0.9,
+                              temperature=1.0,
                               return_prime=False,
                               verbose=False)
     
@@ -176,7 +176,7 @@ def Harmonize_Melody(input_src_midi,
     
     #===============================================================================
 
-    print('Harmonized', len(song), 'out of', len(mel_pitches), 'notes')
+    print('Harmonized', len(song) // 2, 'out of', len(mel_pitches), 'notes')
 
     print('Done!')
     print('=' * 70)
@@ -438,7 +438,7 @@ if __name__ == "__main__":
         gr.Markdown("## Select harmonization options")
 
         source_melody_transpose_value = gr.Slider(-6, 6, value=0, step=1, label="Source melody transpose value", info="You can transpose source melody by specified number of semitones if the original melody key does not harmonize well")
-        model_top_k_sampling_value = gr.Slider(1, 50, value=15, step=1, label="Model sampling top_k value", info="Decreasing this value may produce better harmonization results in some cases")
+        model_top_k_sampling_value = gr.Slider(1, 50, value=25, step=1, label="Model sampling top_k value", info="Decreasing this value may produce better harmonization results in some cases")
         texture_harmonized_chords = gr.Checkbox(label="Texture harmonized chords", value=True, info="Texture harmonized chords for more pleasant listening")
         melody_MIDI_patch_number = gr.Slider(0, 127, value=40, step=1, label="Source melody MIDI patch number")
         harmonized_accompaniment_MIDI_patch_number = gr.Slider(0, 127, value=0, step=1, label="Harmonized accompaniment MIDI patch number")
@@ -467,7 +467,7 @@ if __name__ == "__main__":
 
         gr.Examples(
             [
-            ["USSR Anthem Seed Melody.mid", 0, 15, True, 40, 0, 35],
+            ["USSR Anthem Seed Melody.mid", 0, 25, True, 40, 0, 35],
             ],
             [input_src_midi,
             source_melody_transpose_value,