Update app.py
Browse files
app.py
CHANGED
|
@@ -318,8 +318,24 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
|
|
| 318 |
with gr.Tab("Preprocessors"):
|
| 319 |
with gr.Row(elem_classes="row-main"):
|
| 320 |
with gr.Column(scale=1, elem_id="input_column", elem_classes="input-column"):
|
| 321 |
-
input_image = gr.Image(label="
|
|
|
|
| 322 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 323 |
## TAB LINEART
|
| 324 |
with gr.Tab("Lineart") as tab_lineart:
|
| 325 |
with gr.Row():
|
|
@@ -346,6 +362,8 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
|
|
| 346 |
with gr.Row():
|
| 347 |
run_btn_lineart = gr.Button("Los", variant="primary", visible=True)
|
| 348 |
run_btn_canny = gr.Button("Los", variant="primary", visible=False)
|
|
|
|
|
|
|
| 349 |
|
| 350 |
all_run_btns = [run_btn_lineart, run_btn_canny]
|
| 351 |
|
|
@@ -357,8 +375,9 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
|
|
| 357 |
|
| 358 |
return {"visible": lineart_btn_visible, "__type__": "update"}, {"visible": canny_btn_visible, "__type__": "update"}
|
| 359 |
|
| 360 |
-
|
| 361 |
-
|
|
|
|
| 362 |
|
| 363 |
with gr.Column(scale=2):
|
| 364 |
gallery = gr.Gallery(label="Generated images", show_label=False, interactive=False, format="png", elem_id="output_gallery", elem_classes="output-gallery", columns=[3], rows=[2], object_fit="contain", height="auto", type="filepath")
|
|
@@ -375,7 +394,7 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
|
|
| 375 |
# MARK: Button Runs
|
| 376 |
run_btn_lineart.click(fn=lineart, inputs=[input_image, resolution, preprocessor_name, invert, gallery], outputs=[gallery])
|
| 377 |
run_btn_canny.click(fn=canny, inputs=[input_image, resolution, low_threshold, high_threshold, gallery], outputs=[gallery])
|
| 378 |
-
|
| 379 |
|
| 380 |
"""
|
| 381 |
with gr.Blocks(theme=theme, css="custom.css", js="javascript.js") as demo:
|
|
|
|
| 318 |
with gr.Tab("Preprocessors"):
|
| 319 |
with gr.Row(elem_classes="row-main"):
|
| 320 |
with gr.Column(scale=1, elem_id="input_column", elem_classes="input-column"):
|
| 321 |
+
input_image = gr.Image(label="Input Image", type="numpy", height=512)
|
| 322 |
+
|
| 323 |
|
| 324 |
+
## TAB OPENPOSE
|
| 325 |
+
with gr.Tab("Lineart") as tab_openpose:
|
| 326 |
+
with gr.Row():
|
| 327 |
+
with gr.Column():
|
| 328 |
+
# # input_image = gr.Image(source='upload', type="numpy")
|
| 329 |
+
# input_image = gr.Image(label="Input Image", type="numpy", height=512)
|
| 330 |
+
hand_and_face = gr.Checkbox(label='Hand and Face', value=False)
|
| 331 |
+
resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
|
| 332 |
+
run_button = gr.Button("Run")
|
| 333 |
+
# run_button = gr.Button(label="Run")
|
| 334 |
+
# with gr.Column():
|
| 335 |
+
# gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
|
| 336 |
+
# gallery = gr.Gallery(label="Generated images", show_label=False, height="auto")
|
| 337 |
+
# run_button.click(fn=openpose, inputs=[input_image, resolution, hand_and_face], outputs=[gallery])
|
| 338 |
+
|
| 339 |
## TAB LINEART
|
| 340 |
with gr.Tab("Lineart") as tab_lineart:
|
| 341 |
with gr.Row():
|
|
|
|
| 362 |
with gr.Row():
|
| 363 |
run_btn_lineart = gr.Button("Los", variant="primary", visible=True)
|
| 364 |
run_btn_canny = gr.Button("Los", variant="primary", visible=False)
|
| 365 |
+
run_button_openpose = gr.Button("Los", variant="primary", visible=True)
|
| 366 |
+
|
| 367 |
|
| 368 |
all_run_btns = [run_btn_lineart, run_btn_canny]
|
| 369 |
|
|
|
|
| 375 |
|
| 376 |
return {"visible": lineart_btn_visible, "__type__": "update"}, {"visible": canny_btn_visible, "__type__": "update"}
|
| 377 |
|
| 378 |
+
tab_lineart.select(fn=lambda: set_inputs("tab_lineart"), inputs=None, outputs=[run_btn_lineart, run_btn_canny, run_button_openpose])
|
| 379 |
+
tab_canny.select(fn=lambda: set_inputs("tab_canny"), inputs=None, outputs=[run_btn_lineart, run_btn_canny, run_button_openpose])
|
| 380 |
+
tab_openpose.select(fn=lambda: set_inputs("tab_openpose"), inputs=None, outputs=[run_btn_lineart, run_btn_canny, run_button_openpose])
|
| 381 |
|
| 382 |
with gr.Column(scale=2):
|
| 383 |
gallery = gr.Gallery(label="Generated images", show_label=False, interactive=False, format="png", elem_id="output_gallery", elem_classes="output-gallery", columns=[3], rows=[2], object_fit="contain", height="auto", type="filepath")
|
|
|
|
| 394 |
# MARK: Button Runs
|
| 395 |
run_btn_lineart.click(fn=lineart, inputs=[input_image, resolution, preprocessor_name, invert, gallery], outputs=[gallery])
|
| 396 |
run_btn_canny.click(fn=canny, inputs=[input_image, resolution, low_threshold, high_threshold, gallery], outputs=[gallery])
|
| 397 |
+
run_button_openpose.click(fn=openpose, inputs=[input_image, resolution, hand_and_face], outputs=[gallery])
|
| 398 |
|
| 399 |
"""
|
| 400 |
with gr.Blocks(theme=theme, css="custom.css", js="javascript.js") as demo:
|