Update app.py

app.py

@@ -24,10 +24,14 @@ from midi_to_colab_audio import midi_to_colab_audio
                        
 @spaces.GPU
 def Generate_Rock_Song(input_midi, 
-                       input_freestyle_continuation, 
                        input_number_prime_chords, 
                        input_use_original_durations,
-                       input_match_original_pitches_counts
+                       input_match_original_pitches_counts,
+                       input_freestyle_continuation,
+                       input_number_prime_tokens,
+                       input_number_gen_tokens,
+                       input_model_temperature,
+                       input_model_top_k
                       ):
 
     #===============================================================================
@@ -44,10 +48,14 @@ def Generate_Rock_Song(input_midi,
     print('Requested settings:')
     print('=' * 70)
     print('Input MIDI file name:', fn)
-    print('Freestyle continuation:', input_freestyle_continuation)
     print('Number of prime chords:', input_number_prime_chords)
     print('Use original durations:', input_use_original_durations)
     print('Match original pitches counts:', input_match_original_pitches_counts)
+    print('Freestyle continuation:', input_freestyle_continuation)
+    print('Number of prime tokens:', input_number_prime_tokens)
+    print('Number of tokens to generate:', input_number_gen_tokens)
+    print('Model temperature:', input_model_temperature)
+    print('Model sampling top k value:', input_model_top_k)
     print('=' * 70)
 
     #===============================================================================
@@ -417,7 +425,8 @@ if __name__ == "__main__":
         output_plot = gr.Plot(label="Output MIDI score plot")
         output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
 
-        run_event = run_btn.click(Generate_Rock_Song, [input_number_prime_chords, 
+        run_event = run_btn.click(Generate_Rock_Song, [input_midi,
+                                                       input_number_prime_chords, 
                                                        input_use_original_durations,
                                                        input_match_original_pitches_counts,
                                                        input_freestyle_continuation,
@@ -432,7 +441,8 @@ if __name__ == "__main__":
             [["Sharing The Night Together.kar", 256, True, True, False, 512, 1024, 0.9, 10],
              ["Hotel California.mid", 256, True, True, False, 512, 1024, 0.9, 10],
             ],
-            [input_number_prime_chords, 
+            [input_midi,
+             input_number_prime_chords, 
              input_use_original_durations,
              input_match_original_pitches_counts,
              input_freestyle_continuation,