import gradio as gr import torch from gradio.themes.utils import sizes theme = gr.themes.Default(radius_size=sizes.radius_none).set( block_label_text_color = '#4D63FF', block_title_text_color = '#4D63FF', button_primary_text_color = '#4D63FF', button_primary_background_fill='#FFFFFF', button_primary_border_color='#4D63FF', button_primary_background_fill_hover='#EDEFFF', ) model2 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained=True, progress=False) model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1") face2paint = torch.hub.load('AK391/animegan2-pytorch:main', 'face2paint', size=512,side_by_side=False) def inference(img, ver): if ver == 'version 2 (๐บ robustness,๐ป stylization)': out = face2paint(model2, img) else: out = face2paint(model1, img) return out with gr.Blocks(theme=theme, css="footer {visibility: hidden}") as demo: gr.Markdown("""