app.py

import os
import json
from PIL import Image
from skimage import io
import gradio as gr
import os
import cv2
from modelscope.outputs import OutputKeys
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
import PIL
import numpy as np

img_colorization = pipeline(Tasks.image_colorization, model='iic/cv_ddcolor_image-colorization')
img_path = 'input.png'
##result = img_colorization(img_path)
##cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG])
def color(image):
    output = img_colorization(image[...,::-1])
    result = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
    result = result[...,::-1]
    print('infer finished!')
    return result


title = "老照片修复"
description = "上传图片，达到老照片修复"
examples = [['./input.png'],]

demo = gr.Interface(fn=color,inputs="image",outputs="image",examples=examples,title=title,description=description)

if __name__ == "__main__":
    demo.launch(share=True)