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() | |
with gr.Column(): | |
result = gr.Image(label="Result", elem_id="result") | |
run_button.click( | |
fn=model.predict, | |
inputs=input_image, | |
outputs=result, | |
) | |
if __name__ == "__main__": | |
demo.queue(max_size=20).launch() | |