import gradio as gr | |
from functools import partial | |
from makeanime.cli import main | |
generate_image = partial(main, is_gradio=True) | |
with gr.Blocks() as demo: | |
gr.Markdown("# makeanime: Turn your image into an anime") | |
with gr.Row(): | |
with gr.Column(): | |
image_input = gr.Image(label="Upload your image") | |
prompt_input = gr.Text(label="Prompt") | |
style_weight_slider = gr.Slider( | |
label="Style Weight", minimum=0.0, maximum=1.0, value=0.5, step=0.1 | |
) | |
face_weight_slider = gr.Slider( | |
label="Face Weight", minimum=0.0, maximum=1.0, value=0.5, step=0.1 | |
) | |
generate_button = gr.Button("Generate Anime Image") | |
with gr.Column(): | |
result_output = gr.Image() | |
generate_button.click( | |
fn=generate_image, | |
inputs=[image_input, prompt_input, style_weight_slider, face_weight_slider], | |
outputs=result_output, | |
) | |
if __name__ == "__main__": | |
demo.launch() | |