|
|
|
|
|
import gradio as gr |
|
|
from video_process import add_mask |
|
|
def ui(): |
|
|
|
|
|
with gr.Blocks() as demo: |
|
|
gr.Markdown( |
|
|
"## Hide Face Using Squid Game Masks", |
|
|
elem_classes="text-center" |
|
|
) |
|
|
mask_names=["Front Man Mask", "Guards Mask", "Red Mask", "Blue Mask"] |
|
|
dummy_examples=[["./assets/zuck.mp4"]] |
|
|
with gr.Row(): |
|
|
with gr.Column(): |
|
|
video_input = gr.Video(label="Upload Video") |
|
|
mask_selector = gr.Dropdown(choices=mask_names, label="Select Mask") |
|
|
submit_btn = gr.Button("Apply Mask") |
|
|
|
|
|
with gr.Accordion('Mask Settings', open=False): |
|
|
mask_up = gr.Slider(minimum=0, maximum=100, label="Mask Up (Forehead Side)", value=10) |
|
|
mask_down = gr.Slider(minimum=0, maximum=100, label="Mask Down (Chin Side)", value=0) |
|
|
|
|
|
with gr.Column(): |
|
|
output_video = gr.Video(label="Output Video") |
|
|
download_video = gr.File(label="Download Video") |
|
|
|
|
|
inputs = [video_input, mask_selector, mask_up, mask_down] |
|
|
outputs = [output_video, download_video] |
|
|
|
|
|
submit_btn.click(add_mask, inputs=inputs, outputs=outputs) |
|
|
gr.Examples(examples=dummy_examples, |
|
|
inputs=[video_input] ) |
|
|
|
|
|
return demo |
|
|
import click |
|
|
@click.command() |
|
|
@click.option("--debug", is_flag=True, default=False, help="Enable debug mode.") |
|
|
@click.option("--share", is_flag=True, default=False, help="Enable sharing of the interface.") |
|
|
def main(debug, share): |
|
|
|
|
|
demo=ui() |
|
|
demo.queue().launch(debug=debug, share=share) |
|
|
if __name__ == "__main__": |
|
|
main() |
