Update app.py

app.py

@@ -15,19 +15,15 @@ import matplotlib.pyplot as plt
 
 in_space = os.getenv("SYSTEM") == "spaces"
 
-def run(search_prompt, mid=None, progress=gr.Progress()):
-    
-    if mid == None:
-        
-        for m in progress.tqdm(meta_data):
-            mid_seq = m[1][17:-1]
-            mid_seq_ticks = m[1][16][1]
-            mdata = m[1][:16]
-            break
-       
-    elif mid is not None:
-        mid_seq = MIDI.midi2score(mid)
+def run(midi, progress=gr.Progress()):
 
+    mid_seq = MIDI.midi2score(midi)
+    
+    for m in progress.tqdm(meta_data):
+        mid_seq = m[1][17:-1]
+        mid_seq_ticks = m[1][16][1]
+        mdata = m[1][:16]
+        break
 
     x = []
     y = []
@@ -67,11 +63,6 @@ if __name__ == "__main__":
     
     soundfont_path = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
     meta_data_path = "meta-data/LAMD_META_10000.pickle"
-    
-    models_info = {"generic pretrain model": ["skytnt/midi-model", ""],
-                   "j-pop finetune model": ["skytnt/midi-model-ft", "jpop/"],
-                   "touhou finetune model": ["skytnt/midi-model-ft", "touhou/"]}
-
 
     print('Loading meta-data...')
     with open(meta_data_path, 'rb') as f:
@@ -89,17 +80,8 @@ if __name__ == "__main__":
                     " for faster running and longer generation"
                     )
         
-        with gr.Tabs():
-            with gr.TabItem("instrument prompt") as tab1:
-                
-                search_prompt = gr.Textbox(label="search prompt")
-                
-            with gr.TabItem("midi prompt") as tab2:
-                input_midi = gr.File(label="input midi", file_types=[".midi", ".mid"], type="binary")
 
-        with gr.Accordion("options", open=False):
- 
-            input_allow_cc = gr.Checkbox(label="allow midi cc event", value=True)
+        input_midi = gr.File(label="input midi", file_types=[".midi", ".mid"], type="binary")
             
         search_btn = gr.Button("search", variant="primary")
         
@@ -108,7 +90,7 @@ if __name__ == "__main__":
         output_midi_seq = gr.Textbox(label="output midi metadata")
         output_plot = gr.Plot(label="output midi plot")
         
-        run_event = search_btn.click(run, [search_prompt],
+        run_event = search_btn.click(run, [input_midi],
                                   [output_midi_seq, output_midi, output_audio, output_plot])
         
     app.queue(1).launch(server_port=opt.port, share=opt.share, inbrowser=True)