Update app.py

app.py

@@ -21,7 +21,7 @@ examples = [[os.path.join(examples_dir, img), 0.5] for img in os.listdir(example
 
 
 model = CustomResNet()
-model.load_state_dict(torch.load('modelp.ckpt')['state_dict'], map_location=torch.device('cpu')) #, strict = False)
+model.load_state_dict(torch.load('modelp.ckpt', map_location='cpu')['state_dict']) #, strict = False)
 # model = model.cpu()
 
 classes = ['plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']