new model names

app.py

@@ -82,8 +82,8 @@ def download_model(url, model_path):
             f.write(response.content)
 
 # Model URL and path
-model_urls  = [f'https://huggingface.co/shiyi-li/BirdAST/resolve/main/BirdAST_Baseline_5folds_fold_{i}.pth' for i in range(5)]
-model_paths = [f'BirdAST_Baseline_5folds_fold_{i}.pth' for i in range(5)]
+model_urls  = [f'https://huggingface.co/shiyi-li/BirdAST/resolve/main/BirdAST_Baseline_GroupKFold_fold_{i}.pth' for i in range(5)]
+model_paths = [f'BirdAST_Baseline_GroupKFold_fold_{i}.pth' for i in range(5)]
 
 for (model_url, model_path) in zip(model_urls, model_paths):
     download_model(model_url, model_path)
@@ -99,7 +99,7 @@ for i in range(5):
     eval_models[i].eval()
 
 # Load the species mapping
-label_mapping = pd.read_csv('BirdAST_Baseline_5folds_label_map.csv')
+label_mapping = pd.read_csv('BirdAST_Baseline_GroupKFold_label_map.csv')
 species_id_to_name = {row['species_id']: row['scientific_name'] for index, row in label_mapping.iterrows()}
 
 def preprocess_for_inference(audio_arr, sr):