| | |
| |
|
| | import gradio as gr |
| | import torch |
| |
|
| | from app_canny import create_demo as create_demo_canny |
| | from app_depth import create_demo as create_demo_depth |
| | from app_ip2p import create_demo as create_demo_ip2p |
| | from app_lineart import create_demo as create_demo_lineart |
| | from app_mlsd import create_demo as create_demo_mlsd |
| | from app_normal import create_demo as create_demo_normal |
| | from app_openpose import create_demo as create_demo_openpose |
| | from app_scribble import create_demo as create_demo_scribble |
| | from app_scribble_interactive import create_demo as create_demo_scribble_interactive |
| | from app_segmentation import create_demo as create_demo_segmentation |
| | from app_shuffle import create_demo as create_demo_shuffle |
| | from app_softedge import create_demo as create_demo_softedge |
| | from model import Model |
| | from settings import ALLOW_CHANGING_BASE_MODEL, DEFAULT_MODEL_ID, SHOW_DUPLICATE_BUTTON |
| |
|
| | DESCRIPTION = "# ControlNet v1.1" |
| |
|
| | if not torch.cuda.is_available(): |
| | DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>" |
| |
|
| | model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="Canny") |
| |
|
| | with gr.Blocks(css_paths="style.css") as demo: |
| | gr.Markdown(DESCRIPTION) |
| | gr.DuplicateButton( |
| | value="Duplicate Space for private use", |
| | elem_id="duplicate-button", |
| | visible=SHOW_DUPLICATE_BUTTON, |
| | ) |
| |
|
| | with gr.Tabs(): |
| | with gr.Tab("Canny"): |
| | create_demo_canny(model.process_canny) |
| | with gr.Tab("MLSD"): |
| | create_demo_mlsd(model.process_mlsd) |
| | with gr.Tab("Scribble"): |
| | create_demo_scribble(model.process_scribble) |
| | with gr.Tab("Scribble Interactive"): |
| | create_demo_scribble_interactive(model.process_scribble_interactive) |
| | with gr.Tab("SoftEdge"): |
| | create_demo_softedge(model.process_softedge) |
| | with gr.Tab("OpenPose"): |
| | create_demo_openpose(model.process_openpose) |
| | with gr.Tab("Segmentation"): |
| | create_demo_segmentation(model.process_segmentation) |
| | with gr.Tab("Depth"): |
| | create_demo_depth(model.process_depth) |
| | with gr.Tab("Normal map"): |
| | create_demo_normal(model.process_normal) |
| | with gr.Tab("Lineart"): |
| | create_demo_lineart(model.process_lineart) |
| | with gr.Tab("Content Shuffle"): |
| | create_demo_shuffle(model.process_shuffle) |
| | with gr.Tab("Instruct Pix2Pix"): |
| | create_demo_ip2p(model.process_ip2p) |
| |
|
| | with gr.Accordion(label="Base model", open=False): |
| | with gr.Row(): |
| | with gr.Column(scale=5): |
| | current_base_model = gr.Text(label="Current base model") |
| | with gr.Column(scale=1): |
| | check_base_model_button = gr.Button("Check current base model") |
| | with gr.Row(): |
| | with gr.Column(scale=5): |
| | new_base_model_id = gr.Text( |
| | label="New base model", |
| | max_lines=1, |
| | placeholder="stable-diffusion-v1-5/stable-diffusion-v1-5", |
| | info="The base model must be compatible with Stable Diffusion v1.5.", |
| | interactive=ALLOW_CHANGING_BASE_MODEL, |
| | ) |
| | with gr.Column(scale=1): |
| | change_base_model_button = gr.Button("Change base model", interactive=ALLOW_CHANGING_BASE_MODEL) |
| | if not ALLOW_CHANGING_BASE_MODEL: |
| | gr.Markdown( |
| | """The base model is not allowed to be changed in this Space so as not to slow down the demo, but it can be changed if you duplicate the Space.""" |
| | ) |
| |
|
| | check_base_model_button.click( |
| | fn=lambda: model.base_model_id, |
| | outputs=current_base_model, |
| | queue=False, |
| | api_name="check_base_model", |
| | ) |
| | gr.on( |
| | triggers=[new_base_model_id.submit, change_base_model_button.click], |
| | fn=model.set_base_model, |
| | inputs=new_base_model_id, |
| | outputs=current_base_model, |
| | api_name=False, |
| | concurrency_id="main", |
| | ) |
| |
|
| | if __name__ == "__main__": |
| | demo.queue(max_size=20).launch() |
| |
|
