Update app.py

app.py

@@ -21,7 +21,7 @@ from midi_to_colab_audio import midi_to_colab_audio
 # =================================================================================================
                        
 @spaces.GPU
-def GenerateSong(input_melody_seed_number):
+def Generate_Rock_Song(input_melody_seed_number):
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     start_time = reqtime.time()
@@ -204,8 +204,15 @@ if __name__ == "__main__":
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Generate unique rock music compositions with source augmented RoPE music transformer</h1>")
         gr.Markdown(
             "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Guided-Rock-Music-Transformer&style=flat)\n\n")
+
+        gr.Markdown("## Upload your MIDI or select a sample example MIDI below")
+        
+        input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
         
+        gr.Markdown("## Select generation options")
+
         input_melody_seed_number = gr.Slider(0, 203664, value=0, step=1, label="Select seed melody number")
+        input_find_best_match = gr.Checkbox(label="Find best match", value=False)
 
         run_btn = gr.Button("generate", variant="primary")
 
@@ -217,7 +224,19 @@ 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(GenerateSong, [input_melody_seed_number],
+        run_event = run_btn.click(Generate_Rock_Song, [input_melody_seed_number],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
+        gr.Examples(
+            [["Sharing The Night Together.kar", 0, True], 
+            ],
+            [input_midi, 
+             input_melody_seed_number, 
+             input_find_best_match,
+            ],
+            [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
+            Generate_Rock_Song,
+            cache_examples=False,
+        )
+        
         app.queue().launch()