Spaces:
Sleeping
Sleeping
File size: 676 Bytes
022b7ae d9847a8 022b7ae d9847a8 724eb2f d9847a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
def inference(img):
out = face2paint(model, img)
return out
iface = gr.Interface(fn=inference, inputs=gr.inputs.Image(type="pil"), outputs=gr.outputs.Image(type="pil"),examples=['/content/gongyoo.jpeg'])
iface.launch()
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',
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() |