Spaces:
Running
on
Zero
Running
on
Zero
#!/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() | |