import torchvision from torchvision.models import resnet18 model = resnet18(weights=None) model_weights_path = 'resnet18_flower_classification.pth' model.load_state_dict(torch.load(model_weights_path)) model.eval()