Spaces:
Runtime error
Runtime error
import cv2 | |
from modelscope.outputs import OutputKeys | |
from modelscope.pipelines import pipeline | |
from modelscope.utils.constant import Tasks | |
img_cartoon = pipeline(Tasks.image_portrait_stylization, | |
model='damo/cv_unet_person-image-cartoon_compound-models') | |
result = img_cartoon('input.png') | |
cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG]) | |
print('finished!') | |