Spaces:
Build error
Build error
| import random_face | |
| import gradio as gr | |
| def mobileface(truncate, alpha): | |
| engine = random_face.get_engine() | |
| face = engine.get_random_face(truncate=truncate, alpha=alpha) | |
| return face[:,:,::-1] | |
| inputs = [ | |
| gr.inputs.Checkbox(label="Truncate"), | |
| gr.inputs.Slider(minimum=0, maximum=1, step=None, default=0.5, label="Alpha") | |
| ] | |
| outputs = gr.outputs.Image(type='numpy', label="Output Image") | |
| title = "MobileStyleGAN" | |
| description = "Gradio demo for MobileStyleGAN: A Lightweight Convolutional Neural Network for High-Fidelity Image Synthesis. To use it, simply click submit and optionally adjust alpha and truncation values. Read more at the links below." | |
| article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2104.04767'>MobileStyleGAN: A Lightweight Convolutional Neural Network for High-Fidelity Image Synthesis</a> | <a href='https://github.com/bes-dev/MobileStyleGAN.pytorch'>Github Repo</a></p>" | |
| gr.Interface(mobileface, inputs, outputs, title=title, description=description, article=article).launch() |