import gradio as gr
from huggingface_hub import hf_hub_url, cached_download


def load_model():
    REPO_ID = "MalchuL/JJBAGAN"
    FILENAME = "198_jjba_8_k_2_099_ep.onnx"

    global model

    model = cached_download(
        hf_hub_url(REPO_ID, FILENAME)
    )

    return model

def inference(img):

    return img


title = "JJStyleTransfer"
description = "Gradio Demo for JoJo Bizzare Adventures 5 season style transfer. To use it, simply upload your image, or click one of the examples to load them."
article = "Github Repo Pytorch "
examples = [['demo/karin.jpg'],
            ['demo/tucker.png'],
            ['demo/biden.jpg']]

demo = gr.Interface(
    fn=inference,
    inputs=[gr.inputs.Image(type="pil")],
    outputs=gr.outputs.Image(type="pil"),
    title=title,
    description=description,
    article=article,
    examples=examples)

demo.launch()