Jigsaw net model to unscramble the image

Hi Everyone,
Is there any model for unscrambling the scrambled image.

Task: Given an scrambled image convert it to scrambled image or solve the jigsaw puzzle . Here is one of the repo which does this Jigsaw-Net/README.md at master · anilsathyan7/Jigsaw-Net · GitHub . Does huggingface has one of the model which does similar thing.

This will be a image to image model. Thanks in Advance.