Update app.py

app.py

@@ -135,12 +135,31 @@ def main():
     #sample_file = 'Preludes 2 Through Major keys 39.mid'
     out_file = 'output.mid'
     uploaded_file = st.file_uploader("Sube un archivo MIDI")
+    
+    model=''
+    pesos=''
+    
+    option = st.selectbox(
+    "Elige con que modelo entrenar",
+    ("Maestro", "Lakh"))
 
-    if uploaded_file is not None:
-
+    
+    
+    if uploaded_file and option is not None:
+
+        if option="Maestro":
+            model="mi_modelo_music.h5"
+            pesos="mi_pesos_music.h5"
+        else:
+            model="mi_modelo03_music.h5"
+            pesos="mi_pesos03_music.h5"
+        
         st.subheader("Archivo cargado:")
         st.write(uploaded_file.name)
 
+        st.subheader("Modelo elegido:")
+        st.write(option)
+        
         # Guardar el archivo en una ubicación temporal
         with open(uploaded_file.name, 'wb') as f:
             f.write(uploaded_file.getbuffer())
@@ -152,7 +171,7 @@ def main():
             model = keras.models.load_model("mi_modelo_music.h5")
     
         model.load_weights("mi_pesos_music.h5", skip_mismatch=False, by_name=False, options=None)
-    
+        
         # Convertir MIDI generado por el modelo a archivo WAV
         pm = pretty_midi.PrettyMIDI(sample_file)
         instrument_name = pretty_midi.program_to_instrument_name(pm.instruments[0].program)