Update app.py

app.py

@@ -45,8 +45,8 @@ 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_best.pt', map_location='cpu')
-model.load_state_dict(state_dict)  # Load your trained weights
+model.load_state_dict(torch.load('densenet169_seed40_best.pt', map_location='cpu'))
+ # Load your trained weights
 model.eval()
 
 # ✅ Class Names