new model names

app.py

@@ -90,7 +90,7 @@ for (model_url, model_path) in zip(model_urls, model_paths):
 
 # Load the model (assumes you have the model architecture defined)
 eval_models = [BirdAST(Config().backbone_name, Config().n_classes, n_mlp_layers=1, activation='silu') for i in range(5)]
-state_dicts = [torch.load(f'BirdAST_Baseline_5folds_fold_{i}.pth', map_location='cpu') for i in range(5)]
+state_dicts = [torch.load(f'BirdAST_Baseline_GroupKFold_fold_{i}.pth', map_location='cpu') for i in range(5)]
 for idx, sd in enumerate(state_dicts):
     eval_models[idx].load_state_dict(sd)