File size: 435 Bytes
69a9435 157f1bb d306302 157f1bb d306302 157f1bb d306302 157f1bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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()
|