import gradio as gr
from transformers import pipeline


model_name = "joeddav/xlnet-base-cased"
waifu2x = pipeline("image-upscaling", model=model_name)


def image_upscaler(input_image):
    output_image = waifu2x(input_image)[0]['generated']
    return output_image


input_image = gr.inputs.Image(label="Input Image")
output_image = gr.outputs.Image(label="Enlarged Image", type="numpy")

title = "Waifu2x Image Upscaler"
description = "This app uses the Hugging Face transformers library to upscale images using the Waifu2x model."

gr.Interface(fn=image_upscaler, inputs=input_image, outputs=output_image, title=title, description=description).launch()