Update app.py

app.py

@@ -9,7 +9,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 new_model = models.resnet18()
 num_ftrs = new_model.fc.in_features 
 new_model.fc = nn.Linear(num_ftrs, 75)
-checkpoint = torch.load('model_best_checkpoint.pth.tar')
+checkpoint = torch.load('model_best_checkpoint.pth.tar', map_location=torch.device('cpu'))
 new_model.load_state_dict(checkpoint['model'])
 new_model.to(device)