Update app.py

app.py

@@ -25,7 +25,7 @@ in_space = os.getenv("SYSTEM") == "spaces"
 # =================================================================================================
 
 @spaces.GPU
-def ClassifyMIDI(input_midi):
+def ClassifyMIDI(input_midi, input_sampling_resolution):
 
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -125,7 +125,7 @@ def ClassifyMIDI(input_midi):
     
     input_data = []
     
-    for i in range(0, len(score)-chunk_size, chunk_size // classification_sampling_resolution):
+    for i in range(0, len(score)-chunk_size, chunk_size // input_sampling_resolution):
         schunk = score[i:i+chunk_size]
     
         if len(schunk) == chunk_size:
@@ -146,7 +146,7 @@ def ClassifyMIDI(input_midi):
     classification_summary_string = '=' * 70
     classification_summary_string += '\n'
 
-    samples_overlap = 340 - chunk_size // classification_sampling_resolution // 3
+    samples_overlap = 340 - chunk_size // input_sampling_resolution // 3
     
     print('Composition has', notes_counter, 'notes')
     print('=' * 70)
@@ -377,6 +377,7 @@ if __name__ == "__main__":
         gr.Markdown("## Upload any MIDI to classify")
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
+        input_sampling_resolution = gr.Slider(1, 5, value=2, step=1, label="Classification sampling resolution")
 
         run_btn = gr.Button("classify", variant="primary")
 
@@ -384,7 +385,7 @@ if __name__ == "__main__":
 
         output_midi_cls_summary = gr.Textbox(label="MIDI classification results")
 
-        run_event = run_btn.click(ClassifyMIDI, [input_midi],
+        run_event = run_btn.click(ClassifyMIDI, [input_midi, input_sampling_resolution],
                                   [output_midi_cls_summary])
 
         app.queue().launch()