refactor: methods

This commit is contained in:
Artemy 2023-04-18 13:16:16 +03:00
parent 3f1d13c1c0
commit 09346cb114
6 changed files with 35 additions and 24 deletions

View file

@ -1,18 +1,20 @@
import gradio as gr
from processing.stacking import stacking
from methods.stack_methods import methods
with gr.Blocks() as app:
gr.Markdown("Stacking images.")
with gr.Row():
with gr.Column():
directory = gr.Text(
placeholder="A directory with many images of the same size", lines=1, label="Directory")
methods = gr.Dropdown(
choices=["denoise", "startracks", "noise extractor", "untrack"], value="denoise", label="Method")
placeholder="A directory with many images of the same size",
lines=1,
label="Directory",
)
method = gr.Dropdown(choices=methods, value=methods[0], label="Method")
submit = gr.Button("Submit")
with gr.Column():
output = gr.Gallery()
submit.click(stacking, inputs=[
directory, methods], outputs=[output])
submit.click(stacking, inputs=[directory, method], outputs=[output])