fixing cuda error

app.py

@@ -105,7 +105,7 @@ to_device(model, device);
 
 # loading the model
 ckp_path = 'indianFood-resnet50.pth'
-model.load_state_dict(torch.load(ckp_path), map_location={'cuda': 'cpu'})
+model.load_state_dict(torch.load(ckp_path, map_location=torch.device('cpu')))
 model.eval()