Update app.py

app.py

@@ -62,7 +62,14 @@ def format_table_data(data_string):
 
 #==========================================================================================================
 
-def ID_MIDI(input_midi):
+
+MODELS = {'Llama 3.3 70B Instruct': 'meta-llama/Llama-3.3-70B-Instruct', 
+            'Mistral Nemo Instruct 2407': 'mistralai/Mistral-Nemo-Instruct-2407'
+            }
+
+#==========================================================================================================
+
+def ID_MIDI(input_midi, input_model):
     
     print('*' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -83,6 +90,7 @@ def ID_MIDI(input_midi):
     print('=' * 70)
     print('Input MIDI file name:', fn)
     print('Input MIDI md5 hash', input_midi_md5hash)
+    print('Input model:', input_model)
     print('=' * 70)
     print('Processing MIDI...Please wait...')
     
@@ -156,7 +164,7 @@ def ID_MIDI(input_midi):
 
     completion = client.chat.completions.create(
         #model="Qwen/Qwen2.5-72B-Instruct",
-        model="mistralai/Mistral-Nemo-Instruct-2407",
+        model=MODELS[input_model],
         messages=messages, 
         max_tokens=500
     )
@@ -228,6 +236,7 @@ if __name__ == "__main__":
         gr.Markdown("## Upload your MIDI")
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"], type="filepath")
+        input_model = gr.Dropdown(['Llama 3.3 70B Instruct', 'Mistral Nemo Instruct 2407'], value='Llama 3.3 70B Instruct', label='Select model')
 
         submit = gr.Button("Identify MIDI", variant="primary")
 
@@ -239,7 +248,8 @@ if __name__ == "__main__":
         output_midi_path_str = gr.Textbox(label="Original MIDI raw path string")
         output_MIDID_results_table = gr.Dataframe(label="MIDID database results table", wrap=True, col_count=(3, 'dynamic'))
         
-        run_event = submit.click(ID_MIDI, [input_midi,                                         
+        run_event = submit.click(ID_MIDI, [input_midi,
+                                           input_model
                                               ],
                                                 [output_midi_md5,
                                                  output_midi_records_count,