(self, args)
| 222 | mask = load_image(mask_url).convert("RGB") |
| 223 | |
| 224 | def __init__(self, args): |
| 225 | super().__init__(args) |
| 226 | self.image = self.image.resize(RESOLUTION_MAPPING[args.ckpt]) |
| 227 | self.mask = self.mask.resize(RESOLUTION_MAPPING[args.ckpt]) |
| 228 | |
| 229 | def run_inference(self, pipe, args): |
| 230 | _ = pipe( |