Update app.py

app.py

@@ -45,7 +45,7 @@ from densenet_withglam import get_model_with_attention
 
 # ✅ Instantiate the model
 model = get_model_with_attention('densenet169', num_classes=3)  # Will have GLAM
-state_dict = torch.load('densenet169_seed40_best2.pt', map_location='cpu')
+state_dict = torch.load('densenet169_seed40_best.pt', map_location='cpu')
 model.load_state_dict(state_dict)  # Load your trained weights
 model.eval()