This code uses Masked Autoencoder of He et al. for impainting a masked region of an image. The demo is available on IPOL, where the mask can be drawn over the image. One can try the models (ViT-Large) trained with the MSE loss or with the GAN loss.
Install packages (PIL, numpy, pytorch, timm) and run:
python main.py --input 000019.jpg --loss GAN
Demo based on the official repo https://github.com/facebookresearch/mae which is under the CC-BY-NC 4.0 license. Check the official repo for more details.