import torch model = torch.load("./model.pt")