model = torchvision.models.densenet121(pretrained=False)
model.load_state_dict(torch.load('best_model.pth'))
-