app.py

#!/usr/bin/env python

from __future__ import annotations

import gradio as gr

from model import Model

DESCRIPTION = '# [MangaLineExtraction_PyTorch](https://github.com/ljsabc/MangaLineExtraction_PyTorch)'

model = Model()

with gr.Blocks(css='style.css') as demo:
    gr.Markdown(DESCRIPTION)
    with gr.Row():
        with gr.Column():
            input_image = gr.Image(label='Input', type='numpy')
            run_button = gr.Button(value='Run')
        with gr.Column():
            result = gr.Image(label='Result', type='numpy', elem_id='result')
    run_button.click(fn=model.predict, inputs=input_image, outputs=result)
demo.queue().launch()