| import gradio as gr | |
| from transformers import pipeline | |
| model_name = "waifu2x-multi-lang" | |
| waifu2x = pipeline("image-upscaling", model=model_name) | |
| def waifu2x_enlarge(input_image): | |
| output_image = waifu2x(input_image) | |
| return output_image | |
| input_image = gr.inputs.Image(label="Input Image") | |
| output_image = gr.outputs.Image(label="Enlarged Image", type="numpy") | |
| gr.Interface(waifu2x_enlarge, input_image, output_image).launch() | |