Update app.py

app.py

@@ -34,6 +34,7 @@ model.fc = nn.Linear(model.fc.in_features, 1000)
 model.to(device)  # Move model to CPU (even if you have a GPU)
 
 checkpoint = torch.load('model.pth', map_location='cpu')
+print(checkpoint.keys())
 
 # Load the model weights
 model.load_state_dict(checkpoint['model_state_dict'], strict=False, map_location=device)