| import fn | |
| import gradio as gr | |
| fn.load_model() | |
| with gr.Blocks() as demo: | |
| title = gr.Markdown('# katanuki') | |
| with gr.Row(): | |
| src_image = gr.Image(label="Source", sources="upload", interactive=True, type="pil") | |
| dst_image = gr.Image(label="Result", interactive=False, type="numpy") | |
| src_image.change( | |
| fn=fn.animeseg, | |
| inputs=[src_image], | |
| outputs=[dst_image], | |
| ) | |
| if __name__ == '__main__': | |
| demo.launch() | |