def __init__(self):
    super(Colorization, self).__init__()
    model = create_model(opt)
    model.setup(opt)
    model.eval()
    self.model = model
def __init__(self):
    super(Colorization, self).__init__()
    model = create_model(opt)
    model.setup(opt)
    model.eval()
    self.model = model