import gradio as gr
from PIL import Image
import torch
model = torch.hub.load(
    "AK391/animegan2-pytorch:main",
    "generator",
    pretrained=True)
face2paint = torch.hub.load(
    'AK391/animegan2-pytorch:main', 'face2paint',trust_repo=True,size=512) 
def inference(img):
    out = face2paint(model, img)
    return out
iface = gr.Interface(fn=inference, inputs=gr.Image(type="pil"), outputs=gr.Image(type="pil"),examples=['gongyoo.jpeg'])
iface.launch()