Update app.py

app.py

@@ -44,7 +44,7 @@ model = models.densenet169(pretrained=False)
 model.classifier = nn.Linear(model.classifier.in_features, 3)
 
 # 2️⃣ Load checkpoint
-checkpoint = torch.load('densenet169_seed40_best.pt', map_location='cpu')
+checkpoint = torch.load('densenet169_seed40_best2.pt', map_location='cpu')
 state_dict = checkpoint['state_dict']
 
 # 3️⃣ Fix state dict