Fix predict_and_save function call - add required model parameter

model_inference.py

@@ -34,6 +34,7 @@ def inferir_basic_pitch(input_file: str) -> str:
         
         # Realizar predicción y guardar archivo MIDI
         predict_and_save(
+            model_or_model_path=ICASSP_2022_MODEL_PATH,
             audio_path_list=[input_file],
             output_directory=BASE_MIDI_DIR,
             save_midi=True,
@@ -43,13 +44,23 @@ def inferir_basic_pitch(input_file: str) -> str:
         )
         
         # Basic Pitch genera el archivo con un nombre específico
-        # Necesitamos encontrar el archivo MIDI generado
+        # El archivo se genera con el nombre del archivo original + "_basic_pitch.mid"
+        base_name = os.path.splitext(os.path.basename(input_file))[0]
+        expected_midi_name = f"{base_name}_basic_pitch.mid"
+        expected_midi_path = os.path.join(BASE_MIDI_DIR, expected_midi_name)
+        
+        # Verificar si se generó el archivo esperado
+        if os.path.exists(expected_midi_path):
+            print(f"Archivo MIDI generado: {expected_midi_path}")
+            return expected_midi_path
+        
+        # Si no está con el nombre esperado, buscar cualquier archivo .mid generado recientemente
         generated_files = [f for f in os.listdir(BASE_MIDI_DIR) if f.endswith('.mid')]
         if generated_files:
             # Tomar el archivo más reciente
             latest_file = max([os.path.join(BASE_MIDI_DIR, f) for f in generated_files], 
                             key=os.path.getctime)
-            print(f"Archivo MIDI generado: {latest_file}")
+            print(f"Archivo MIDI encontrado: {latest_file}")
             return latest_file
         else:
             print("No se generó ningún archivo MIDI")