Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 from torchvision import datasets, models, transforms
 from PIL import Image
 
-LABELS = ['VW Up!', 'Fiat 500']
+LABELS = ['Fiat 500', 'VW Up!']
 
 model = models.resnet18(pretrained=True)
 num_ftrs = model.fc.in_features
@@ -11,7 +11,7 @@ num_ftrs = model.fc.in_features
 model.fc = torch.nn.Linear(num_ftrs, 2)
 
 state_dict = torch.load('up500Model.pt', map_location='cpu')
-model.load_state_dict(state_dict, strict=False)
+model.load_state_dict(state_dict)
 model.eval()
 
 imgTransforms = transforms.Compose([