import os.path as osp

import cv2
from modelscope.outputs import OutputKeys
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks

image_skychange = pipeline(Tasks.image_skychange,
                           model='iic/cv_hrnetocr_skychange')
result = image_skychange(
    {'sky_image': '~/Downloads/sky_image.jpg',
     'scene_image': '/opt/data/face/NXEo0zusSaNB2fa232c84898e92ff165e2dfee59cb54.jpg'})
cv2.imwrite('~/Downloads/result.png',
            result[OutputKeys.OUTPUT_IMG])
print(f'Output written to {osp.abspath("result.png")}')