Update app.py

app.py

@@ -24,11 +24,22 @@ in_space = os.getenv("SYSTEM") == "spaces"
 # =================================================================================================
                        
 @spaces.GPU
-def GenerateSong(input_melody_seed_number):
+def GenerateBridge(input_midi, input_start_note):
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     start_time = reqtime.time()
 
+
+    print('=' * 70)
+
+    fn = os.path.basename(input_midi.name)
+    fn1 = fn.split('.')[0]
+
+    print('-' * 70)
+    print('Input file name:', fn)
+    print('Start note', input_start_note)
+    print('-' * 70)
+
     print('Loading model...')
 
     SEQ_LEN = 3074
@@ -208,8 +219,11 @@ if __name__ == "__main__":
         gr.Markdown(
             "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Bridge-Music-Transformer&style=flat)\n\n")
         
-        input_melody_seed_number = gr.Slider(0, 203664, value=0, step=1, label="Select seed melody number")
-
+        gr.Markdown("## Upload your MIDI or select a sample example MIDI below")
+        gr.Markdown("### Please note that the MIDI must have at least 615 notes for this demo to work")
+        
+        input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
+        input_start_note = gr.Slider(0, 205, value=0, step=1, label="Start note number")
         run_btn = gr.Button("generate", variant="primary")
 
         gr.Markdown("## Generation results")
@@ -220,7 +234,7 @@ 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(GenerateBridge, [input_midi, input_start_note],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         app.queue().launch()