import gradio as gr
from transformers import pipeline
from PIL import Image

# AnimeGANv2 모델 로드
model = pipeline("image-to-image", model="akhaliq/AnimeGANv2")

def transform_image(image):
    result = model(image)
    return result

iface = gr.Interface(
    fn=transform_image,
    inputs=gr.inputs.Image(type="pil"),
    outputs=gr.outputs.Image(type="pil")
)

iface.launch()